Modified the EpidemicEvent class