Minor code reformat.

What can I say, I tend to OCD over these things.
Removes duplicate code.
Merge pull request #4 from potherca/patch-1

Minor code reformat.