Updates to the script