Changes for KNN run