Added unit test to test suite