How to draw fig 2

[JunMa11]

Hi @lukfischer ,

Thanks for your awesome work.

Would it be possible for you to share the code to draw Fig 2?

https://www.nature.com/articles/s41597-020-00608-w#Fig2