Changes to design to make it better for smaller screens