Now with nicer display of the tree