optimize .save and .load