Got the update function working in the new style.