Add Augment abstract class