Make menu toggle work on mobile size