Media query styling fix