Treeview improvements