Using AI to diagnose medical problems