Add restart button.