Add enhanced MPI support