replaced treeViewModel with KnnViewModel