This file could be merged into `src/python/plot.py`, which is maybe a more natural spot for it. Alternatively you could make it a script! I would suggest doing one or the other so we don't create too many small files that will get confusing later.