add the training process