add plot_coef() function