Corrected cooling_update calls and cooling_struct_restore to work for cooling models other than EAGLE