[2/2] Add layout for different screen size and mobile