Update Trainer.h