Add another TODO, this time for the count method