check_balance function

Check covariate balance of the control and treated groups