Made mpi4py optional