Implemented KNN approach