update training script