Add upgrade()