plot_ate function

Plot estimated treatment effects