modifying KNN, I dont think a class is necessary