Modified EpidemicSimulator to handle ESimulationStage as Flags