Updates to Stokes solver