calculate_ate function

Calculate Average Treatment Effect