Minor modification to tree data visualization