Add count test method