inserted edit stage  into chat flow