you can use  update_stage_id and remove move_to_stage from the model