Add upgrade mode