Fix data_generator.py command line