Tweaked the script.