Modify heap.groovy to display tree correctly