Add tests for ->count(), ->sum()