Improve styling on mobile