Add restart button
Add restart handler