Minor tweaks to mpi.py demo.