Add class to hide things on small screens

Update header to work on small screens