Add upgrade step