Updated test data generator argument parser