Add plot_function