Added wrapper for KNN