main script updated
Merge pull request #1 from anthanh/master

main script updated