Updating the styles to work better on mobile