Add a per-run setup function