update models for 5. Diagnosis and 6. Prognosis