update the script
Merge pull request #1 from zhaoace/develop

update the script