Suggestion for cleaner code

That should really be the default, but, you know, "R".
Merge pull request #9 from statshero/patch-1

Suggestion for cleaner code