Add infection probability calculation to the model, and introduce it into the model output in a rudimentary form.