Working on figure 2