Add methods to Augment