Layout improvements for small screens