Add ability to set the sample size for simulations.