Added Treatment persistence