Added basic unit test.