Update data_stage.r