modify KNN.py