Have a look at the website at: https://codatmo.github.io/Simple_SIR/ It has pointers to this repo and all relevant info.