Update data_parallelizer.py