Add a figure for examining treatment heterogeneity (residuals vs fits split by treatment)

[LucyMcGowan]

r-causal/halfmoon#2

[malcolmbarrett]

@LucyMcGowan you can also do this in halfmoon now: https://r-causal.github.io/halfmoon/reference/plot_stratified_residuals.html