The notebooks folder contains code for replicating the analyses and figures reported in the paper: demo.ipynb: code for generating Figure 1