Wrapping the Flexbox class in a media query so that it only works on larger screens