Added a bit of media-query