Updated save and load for models