Added functionality for MPI implementation