Update knn.py