modiefied knn.py