js fix for mobile menu