Working in the new code of treeview