Make README illustration code a little more fluid

This should make the example code a little more connected.