Added upgrade step
Added upgrade step