Added Unit Test for KNN