Css'ten anlıyor musun bilmiyorum ama kapsayıcıya flex-wrap verip içerisindeki kutuların genişliğini ayarlayıp flex-direction:column yaparsan senin yapıya döner.