Further development on the treeview