[Tms]: Added upgrade