Create kNN.py