Update kdtree.py