Quick self-promo, you can run the simulation and play around with it (and the rest of the repo) in Deepnote: https://beta.deepnote.org/launch?template=python_3.6&url=htt... (Disclaimer: I work there.)