Add graph figure printing to ds gen
Add fancy demonstration procedure for validation