Update Trainer.java