Mobile menu behavior fix and improvements