add function code to EpidemicEvent class