fixed stratify on propensity scores