updates to tree viz