Refactor with helper method
Merge pull request #3 from patbaker82/refactor

Refactor with helper method