updated stokes-balance