Trying to make get_training_set to work.