Now correctly subsetting data when asking for U and (kl)