Add media queries for small screens