Update the upgrade procedure