updated figure 2 part of script