Abstract prob_death_post_tx into separate function

[yiwen-h]

See comment on PR #141: the calculation for the prob_death_post_tx is quite complex (and the code is repeated twice) - would it be better abstracted out as a function into helpers

[yiwen-h]

Don't think this applies anymore given #149