Got the treeview working