Nice simulation! I made my own is javascript which you might be interested in, as it is happens to be pretty similar. Your simulation has more parameters to control tough, and was inspired by a different source (3Blue1Brown's Simulating an Epidemic was my inspiration). If you want to check it out: