Added figure 4.20 code