Add unit tests

Closes #42
Yes, definitely.