KNN almost finished