Change output of extract_training_data to list.