Add upgrade init