Split up plot_fit