Fix issues on smaller screens