tiny fix
add knn