Collapse/Expand all nodes

If I where to call the ExpanAll() (the one that is in the PersonNode class in the DataSet sample) upon a diagram that is collapse it should expand all the collapsed branches right?
And I wanted to Do the opposite I should create a CollapseAll() that might look like this?
Public Overridable Sub CollapseAll() Collapse() For Each node As IGoNode In Me.Destinations Dim child As PersonNode = Nothing If TypeOf node Is PersonNode Then child = CType(node, PersonNode) If Not child Is Nothing Then child.CollapseAll() End If Next End Sub
Should I post this on a separate topic?
Thanks for your help Jake

I haven’t tried it, but it looks like it ought to work.