Updated the script to reflect the changes in the document generation tool