add option to select which variable to plot against