Add menu toggling for small screens