Added new unit test