Improve style on media queries