Small changes to the script