Add flag for parallelization