clean up and add figure 3