Fix mobile menu and more