Change tests so they read data from the training set