Implement KNN

for checking