Fix container styling for small screens