Add optional comm.

Add optional argument comm to initmpi.