Small changes to run simulation with new param