Add restart script