Minor updates to the script.