Add upgrade method