Updates to script