redesign solve_stokes()