Add media-query for better layouting in smaller windows