Some unit tests

Resolves #7.