add UI for new stages