Testing training script