Adds minor refactor
Merge pull request #7 from rjbernaldo/refactor

Adds minor refactor