Implement KNN, simple test case has been applied