Working on treeview directive