Update the script
Merge pull request #2 from VertiPub/wb397

Update the script