added epidemic models