Addressing gender imbalance in AI