Add styling for small screens