Improve styling for mobile