knn localization now works