Adding ability to reset the simulation