Update flow for small screens