kNearestNeighbors working