Adding media queries for small screens