Add unit tests

[fzeiser]

It'd probably be a good idea to add some unit tests to find bugs in the code