AST modification