Added parameter for parallel partition size