Closer to a working version of 2D ADMM.