plot_model function

Plot model fits