Add tests.

100% code coverage.