Update KNN.py