updated augmentations