extend KNN classes