Update the training and test data