Collapse with flex elements - simple approach

Created by: MartijnCuppens

Fixes #22600 (closed)

Simple solution instead of #22758. This solution doesn't add a display property to an element when it's visible.

Demo: https://codepen.io/MartijnCuppens/pen/jzNyyy

cc @Johann-S @mdo @XhmikosR