Add callbacks for update, create, and save