Update dihNearestNeighborConnection.py