This notebook *should* work, with minimal modifications to the imports now: https://github.com/jeromekelleher/sc2ts-paper/blob/main/notebooks/example_arg_analysis.ipynb