Updated StosGrid to use the yield semantic