Text this: Modelling the effect of plant water use traits on yield and stay-green expression in sorghum