Update restart script