change kdtree exercise