added first basic logic for treeview