Add completion for upgrade.