Edit UI flow