public void UpgradeButton() { if (!playerStats) { NewPlayer(); } if (playerStats == null) { return; } if (playerStats.Wood - buildingStats.GetUpgradeWood() >= 0 && playerStats.Stone - buildingStats.GetUpgradeStone() >= 0) { playerStats.Wood -= buildingStats.GetUpgradeWood(); playerStats.Stone -= buildingStats.GetUpgradeStone(); grabPlayerStats.BuildingBoughtSoldRepairedUpgraded(); buildingStats.CalculateRepairCosts(); UpdateRepairCostAndHealth(); buildingStats.BuildingUpgrade(); } }