Added option to restart simulation once its over