chooseUpgrade method update

Wrong number handling