Add media query to detect screen size. Not really working for some reason..