Update knnsearch.py