Add plumbing for AST rewriting.