Refactor script to be more clear
Merge pull request #2 from leejones/lj-refactor-script

Refactor script to be more clear