fixed typo, update figure 4