Improve figure 1