fixed some issues on smaller screens