Add flexbox to layout