use count method to make the test pass