Add multi-processing support