Add code to plot variables