diff --git a/src/main.rs b/src/main.rs index df7538c7d52bcbcab2a233242bf512a14c13fdea..891f303331b359299771382993a94cfa54ff94db 100644 --- a/src/main.rs +++ b/src/main.rs @@ -101,7 +101,13 @@ fn main() -> std::io::Result<()> { let date = date.to_string(); let exponential_estimate = exponential_estimate.to_string(); let gompertz_estimate = gompertz_estimate.to_string(); - writer.write_record(&[date, exponential_estimate, gompertz_estimate]).unwrap(); + let actual_infected = if (day as usize) < days.len() { + days[day as usize].infected.to_string() + } else { + "".to_string() + }; + + writer.write_record(&[date, actual_infected, exponential_estimate, gompertz_estimate]).unwrap(); }; if (day as usize) < days.len() {