Add unit tests to increase coverage.

Ref #1