Added the training module.