Tweaks to restart script