code cleaup.
Merge pull request #11 from ryanermita/fix/code-clean-up

code cleaup.