Add back AST transform infrastructure