Add the upgrade command