Extending balance assessment for the generalized propensity score under multiple imputation