more updates for the layout after testing on mobile