A Python implementation of a kd-tree