Balancing covariates in multi-arm trials via adaptive randomization