Place Figure and Text Near Each Other

[Anjiang-Wei]

In the partition section, Figure 4.7, 4.8. 4.9 are a bit too far from their corresponding text. I find it a little bit hard to read the code and the explanation at the same time.