Added a new class for upgrades.