Add save hook