Using AI to help risk management