Updated CSS for better flexbox support on mobile