added figure 4 notebook