Some changes to commit_stage()