Changed plot to infected instead of prevalence