Small fixes

Note that param setting still finicky, so need to set a and b manually.
Important fix here is to correctly record output in model_perf_summary.