Added after_update and after_create callbacks