Added a media query to check