added get_test_set function to gtsrb
corrections to get_test_set