Add upgrade flow.