Added more AST exploration code