adding function to read in test set