Code for Schreiner et al., 2024, Nat. Commun.