KNN Classifier now works