Improve .nav-menu on small screens