Updating script per issue #3
Merge pull request #4 from isubit/master

Updating script per issue #3