Add updateStageData