Add onBeforeSave() function