Update Trainer.cpp