Added a restart button.