Flexbox CSS fixes