Updated with simulation - issue still with getting equilibrium