counterfactual plots dont match, working on it.