Improve parallelization script