Some work on training script