Got solve_A and solve_A_dagger to work (2D).