Implement KNN (lab5)