Improve layout to work on mobile devices