Adjust `save_computed_output!` to only save what I need

[jbisits]

Some things have changed with what needs saving so after I have sorted what we need adjust this function accordingly.