Change the training function