updated stress-balance functions