Add restart button