Fix color figure 2