worked on Flat & StructuredDataSet