MAINT: use a more flexible interpretation of disease data for the log likelihood calculations