Fix styling for mobile