updated UI for model generator, updated data generator