working on data gen