Fix up display on smaller screens