Como deshacer un flexbox con un evento

Question

Tengo un grupo de divs dentro de un contenedor. El código consiste en que al presionar un botón los divs se ordenen con la propiedad de flexbox y al presionar el otro botón se deshaga la propiedad de flexbox. Pero no he podido con el otro botón.
https://code.sololearn.com/WrC2sx6YN2IZ/?ref=app

Answer 1

No sé porque no se me havia ocurrido antes. Con display='none' esa es la solución