change the input
modified KNN/src/knn.py