Updating script to fix variable names in figure 6.