Updating with function linearization almost done!