Carefully read the comments in each ex.py to understand how the code works, especially the MPI parts.