added support for MPI (uses mpi4py)