Modified create_test_set to allow filtering by training data file