adding disease class