Adding in Trainer code.