Clean up mobile menu code