JS fix for mobile submenu