Minor adjustments to update equations, in code and latex.