add media query for max-width: 640px