Add an AST walker