Updates to register, passing the model around as required for jacobian and parameter updates.