improve menu-collapse thingy for small screens