- estimate <- paste0(round(estimate$r0, 2), " (SI = ", estimated_mu,
- " ", tolower(dataset[, 2]), ")"
+ mu_units <- if (dataset[, 2] == "Days") "day(s)" else "week(s)"
+ estimate <- paste0(
+ round(estimate$r0, 2), " (SI = ", estimated_mu, " ", mu_units, ")"