Update kd_tree.h