Adding unit test.