Updated

Small fixes to the script.