C is referred to as counts in the docstring. I think the variable name should change.