def __init__(self, prob, *args, **kwargs):
     self._prob = prob
     TreeEdge.__init__(self, *args, **kwargs)