Private count tests added for models

Models tests have been updated to take into account the private flag with private and non-private count tests.