Added some unit tests.

Gives a bit of test coverage.