Make ui better for sm and xs screens.