add kNN.py