Clean up

Thanks @michaelbollig
Merge pull request #3 from jonom/pr/2

Clean up