Improve CSS flexbox usage