Update trainer.m