Add upgrade_all