Minor changes.
Minor fix to the script.