Fixed the toggle button for menu on small screens
Updated JS for toggle button