updated the training module.