changed the population affected to be based on the @infected rather then @healthy pop