Added simple unit test.