Added new unit test

Added new unit test