Improved root finding in find_equilibrium.