Added upgrade step