Add augmentation() method