added data_dir to data generator example