start fixing compute_equilibrium