marginal_effects function

Marginal treatment effects