Allow augmented Hessian to be used. Unsure if it is better?