Adding the upgradescreen class itself