add code to impute using KNN