added training script