Revert "Additional cleanup"

This reverts commit 53f796543133e3cdda8f1207b7132bf5cfed2abb.
Merge pull request #4 from james-marsh/master

Revert "Additional cleanup"