Add afterSave()