Updates to kNN for partial distance