Python Kd Tree Search