Add JavaScript to handle menu on mobile