Implemented more of Trainer class