Fix knn.py