minor improvement to plot