training_data_small added