Update to be more idiomatic
Merge pull request #3 from dsyme/patch-2

Update to be more idiomatic