Add  unit tests.