Test MPI code