Updated for better figure 5