Handle upgrade logic