initial with knn.py