Added test for count method.