Small update in the script