Added unit test.
Added unit test.