preparing to refactor of transition prob
some fixes, this now works on the commited example, still having issues
with constants