Add upgrade checks