Add training script