Take default probability as argument.
Changed return variable name.
Filled in function skeletons. Figure out what to do with zero_count_prob.
Don't know.