Update KNN_img.py