Adding new fields for treatments and cures to models