update knn_book.py