Fixes #1 we now have unit tests!