added tests for count method