Update name of extract_training_data function.