Making the menu a bit better on mobile