fixed js and added some styling

wasn't display nav menu correctly on small screens