diff --git a/analysis/dataset_definition.py b/analysis/dataset_definition.py
index 2438d7e..76a4271 100644
--- a/analysis/dataset_definition.py
+++ b/analysis/dataset_definition.py
@@ -122,7 +122,7 @@ def preceding_registration(dx_date):
 )  
 
 # List of diseases and codelists to cycle through
-diseases = ["multiple_sclerosis"]
+diseases = ["multiple_sclerosis", "heart_failure", "ulcerative_colitis"]
 # diseases = ["asthma", "copd", "chd", "stroke", "heart_failure", "dementia", "multiple_sclerosis", "epilepsy", "crohns_disease", "ulcerative_colitis", "dm_type2", "dm_type1", "ckd", "psoriasis", "atopic_dermatitis", "osteoporosis", "hiv", "depression"]
 codelist_types = ["snomed", "ctv", "icd"]