changed the tree visualizer and driver to handle visualizer types