Add support for Augmentation.