add UI for the next step