Fixing the toggle button for mobile