Updates to the script related to prototype injection