Added unit test.