Count issue after rewiring

[hniu1]

Hi,

I have a question about the rewiring. Let's say a->b->c has been rewired as a->b|a->c. The frequency of b->c should subtract the frequency of b|a->c after the rewiring (count[b][c] -= count[b|a][c]). Otherwise, the input frequency and output frequency of b are not equal. I found this issue after running the code.

Thanks