Tweaks to script.