change moduleclass to mpi for mpi4py