Add notes for augmented classes