Add Tree Visualization
Separate ui from core functions