> This was in the test file previously. It is right above the test I
> added for probability_transition.

I was commenting on all this probability stuff. It should go
into a separate pull request as it has nothing to do with heurisch.