remove knn related logic