Improve submenu design on small screens