Updated the automation script.