Added restart button.