Added unit test for count