Update kNN training
add changes