Add an upgrade step