quick refactor of code from identified issue on codeclimate
Merge pull request #13 from interlock/refactor

quick refactor of code from identified issue on codeclimate