add count tests