Add separate max processes for each stage in wrapper class