Refactoring tree_display views

Refactored the correct files this time, hopefully. :)