Added unit test for service