Modeling treatment events in disease progression