Write unit tests

[vandalt]

There should be unit tests for most of the code. Aiming to have this soon after initial release.