ENH: add equalize_channels function