added some visualization for the tree