Switch to K-nearest neighbors