Modify knnv2 code