add optional argument to plot fix