Before revision:  Keeping the flexibility, we can use any machine learning method to learn the conditional mean functions as well as the propensity score.
After revision:  We use machine learning methods to learn the conditional mean functions as well as the propensity score.