Update kNN.py