Functions for propensity scoring, computing avg treatment effect