adds plot_imp and plot_twoway_partial