Tree water model with stem storage and shrinkage