Adding count method for testing purpose