Add after_create and after_update tests