Changed subset generator to return a 2D slice