Improved styling on mobile