Improve design on small screen devices