re-created functionality from previous UI, started getting some treeview stuff working