Switch to k-nearest neighbours