Updating styling for different screen sizes