Further cleaned up flexbox css