add expectation argument to negative binomial simulation code