fix layout bugs on smaller screens