Updating the plots to include the partitioning algorithm

Also, put some captions in math mode and broke up lines.