working on training script