Updated restart script