Added the augment class.