simplified training script