Make more idiomatic
Merge pull request #5 from dsyme/patch-4

Make more idiomatic