corrections on data augmentation