Adding docs to the AST rewriter