Got the tree view working, need to finalize it.