Fix design for small screens
Fix design for small screens