Upgrade function added