Added the disease state model