WIP: in stochastic module, collect parameters in an internal class for easier passing between functions