Using the mpi4py Python module