Added a unit test case