add in the plotting call