Add unit tests.