Python Kd Tree Example