cleanup code
Merge pull request #7 from jlambert121/cleanup

cleanup code