Add upgrade/integrate calls