couple fixes to training_data