Enhanced mobile menu with js