finishing the KNN main.py