Blue - does not have disease Red - has disease and is infectious Black - removed (either dead or immune) play multiple times - you will get different results.
Inspiration from 3blue1brown, I did this in scratch to see if it was possible in a block based language.