Added tests to cover count methods