Expected counts can be computed
small change compute_probability function
init transition dict