Clean up smaller screen layout