Clean up MPI wrapping