Made some more adjustments to the script.