Extended the stack size computation with the new AST node.