ATE by regression, example code in separate file