Add support for @augments (close #8)