Adding AST traversal