rewrote data augmentation, now can read in augmentation parameters