Implement Knn Classifier Python