Update diagnosis.R