Tweaks to update()