Add workload arg to plot_cpu