Include clin_diagnos in imputed dataset