Improve the layout for mobile device