Added treatment type to grid