i.e., the well equations are now considered in the linearization of the global system of equations. TODO: make it work for MPI parallel runs...