Add basic documentation to Trainer