Giving AI penalties to get better diagnoses