Add a plot function to the ModelFit class.