Attempt on fixing treeview