redo layout to look better on small screens