Made changes to nompi4py. Should work now