/**** 
* Classes
****/ 
var ActionButton = Container.expand(function () {
	var self = Container.call(this);
	self.init = function (text, action, x, y) {
		self.action = action;
		// Button graphics
		self.buttonGraphics = self.attachAsset('actionButton', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		// Button text
		self.buttonText = new Text2(text, {
			size: 28,
			fill: 0xFFFFFF
		});
		self.buttonText.anchor.set(0.5, 0.5);
		self.addChild(self.buttonText);
		self.x = x;
		self.y = y;
		return self;
	};
	self.down = function (x, y, obj) {
		if (self.action) {
			self.action();
		}
	};
	return self;
});
// Game state variables
var Territory = Container.expand(function () {
	var self = Container.call(this);
	self.init = function (id, x, y, owner) {
		self.id = id;
		self.owner = owner; // 0 = neutral, 1 = player, 2 = enemy
		self.armies = owner === 0 ? 0 : Math.floor(Math.random() * 3) + 1;
		self.selected = false;
		self.lastSelected = false;
		// Create territory graphics
		self.updateGraphics();
		// Position
		self.x = x;
		self.y = y;
		// Army display
		self.armyText = new Text2(self.armies.toString(), {
			size: 24,
			fill: 0x000000
		});
		self.armyText.anchor.set(0.5, 0.5);
		self.armyText.x = 0;
		self.armyText.y = 0;
		self.addChild(self.armyText);
		return self;
	};
	self.updateGraphics = function () {
		// Remove existing graphics
		if (self.graphics) {
			self.removeChild(self.graphics);
		}
		var assetName = 'neutralTerritory';
		if (self.selected) {
			assetName = 'selectedTerritory';
		} else if (self.owner === 1) {
			assetName = 'playerTerritory';
		} else if (self.owner === 2) {
			assetName = 'enemyTerritory';
		}
		self.graphics = self.attachAsset(assetName, {
			anchorX: 0.5,
			anchorY: 0.5
		});
	};
	self.setOwner = function (newOwner, armies) {
		self.owner = newOwner;
		self.armies = armies || 1;
		self.updateGraphics();
		self.armyText.setText(self.armies.toString());
	};
	self.addArmies = function (count) {
		self.armies += count;
		self.armyText.setText(self.armies.toString());
	};
	self.down = function (x, y, obj) {
		if (gameState === 'playing') {
			selectTerritory(self);
		}
	};
	return self;
});

/**** 
* Initialize Game
****/ 
var game = new LK.Game({
	backgroundColor: 0x2c3e50
});

/**** 
* Game Code
****/ 
// Game state variables
// Territory and game assets
var gameState = 'playing'; // 'playing', 'gameOver', 'victory'
var currentTurn = 1; // 1 = player, 2 = enemy
var selectedTerritory = null;
var territories = [];
var playerTerritories = 0;
var enemyTerritories = 0;
var totalTerritories = 0;
// UI elements
var turnText, scoreText, instructionText, playerArmyText;
var actionButtons = [];
// Game settings
var mapWidth = 6;
var mapHeight = 8;
var territorySpacing = 140;
var startX = 2048 / 2 - mapWidth * territorySpacing / 2 + territorySpacing / 2;
var startY = 400;
// Create the game map
function createMap() {
	var id = 0;
	for (var row = 0; row < mapHeight; row++) {
		for (var col = 0; col < mapWidth; col++) {
			var x = startX + col * territorySpacing;
			var y = startY + row * territorySpacing;
			// Determine initial owner
			var owner = 0; // neutral by default
			if (row === 0 && col === 0) {
				owner = 1; // player starts here
			} else if (row === mapHeight - 1 && col === mapWidth - 1) {
				owner = 2; // enemy starts here
			} else if (Math.random() < 0.15) {
				owner = Math.random() < 0.5 ? 1 : 2; // some random territories
			}
			var territory = new Territory();
			territory.init(id, x, y, owner);
			territories.push(territory);
			game.addChild(territory);
			id++;
		}
	}
	updateTerritoryCount();
}
// Update territory ownership counts
function updateTerritoryCount() {
	playerTerritories = 0;
	enemyTerritories = 0;
	var playerTotalArmies = 0;
	totalTerritories = territories.length;
	for (var i = 0; i < territories.length; i++) {
		if (territories[i].owner === 1) {
			playerTerritories++;
			playerTotalArmies += territories[i].armies;
		} else if (territories[i].owner === 2) {
			enemyTerritories++;
		}
	}
	// Update score display
	if (scoreText) {
		scoreText.setText("Player: " + playerTerritories + " | Enemy: " + enemyTerritories);
	}
	// Update player army display
	if (playerArmyText) {
		playerArmyText.setText("Total Armies: " + playerTotalArmies);
	}
	// Check victory conditions
	if (playerTerritories >= Math.floor(totalTerritories * 0.75)) {
		gameState = 'victory';
		LK.showYouWin();
	} else if (enemyTerritories >= Math.floor(totalTerritories * 0.75)) {
		gameState = 'gameOver';
		LK.showGameOver();
	}
}
// Territory selection
function selectTerritory(territory) {
	if (currentTurn !== 1) return; // Only during player turn
	// Deselect previous territory
	if (selectedTerritory) {
		selectedTerritory.selected = false;
		selectedTerritory.updateGraphics();
	}
	// Select new territory
	if (territory.owner === 1) {
		// Only select player territories
		selectedTerritory = territory;
		territory.selected = true;
		territory.updateGraphics();
		updateInstructions();
	}
}
// Get neighboring territories
function getNeighbors(territory) {
	var neighbors = [];
	var index = territories.indexOf(territory);
	var row = Math.floor(index / mapWidth);
	var col = index % mapWidth;
	// Check all 4 directions
	var directions = [{
		row: -1,
		col: 0
	},
	// up
	{
		row: 1,
		col: 0
	},
	// down
	{
		row: 0,
		col: -1
	},
	// left
	{
		row: 0,
		col: 1
	} // right
	];
	for (var i = 0; i < directions.length; i++) {
		var newRow = row + directions[i].row;
		var newCol = col + directions[i].col;
		if (newRow >= 0 && newRow < mapHeight && newCol >= 0 && newCol < mapWidth) {
			var neighborIndex = newRow * mapWidth + newCol;
			neighbors.push(territories[neighborIndex]);
		}
	}
	return neighbors;
}
// Attack action
function attackTerritory() {
	if (!selectedTerritory || selectedTerritory.armies <= 1) return;
	var neighbors = getNeighbors(selectedTerritory);
	var targets = [];
	for (var i = 0; i < neighbors.length; i++) {
		if (neighbors[i].owner !== 1) {
			targets.push(neighbors[i]);
		}
	}
	if (targets.length > 0) {
		var target = targets[Math.floor(Math.random() * targets.length)];
		// Simple combat resolution
		var attackPower = selectedTerritory.armies - 1;
		var defensePower = target.armies;
		if (attackPower > defensePower) {
			// Victory
			target.setOwner(1, attackPower - defensePower);
			selectedTerritory.armies = 1;
			selectedTerritory.armyText.setText(selectedTerritory.armies.toString());
			LK.getSound('capture').play();
		} else {
			// Defeat
			selectedTerritory.armies = 1;
			selectedTerritory.armyText.setText(selectedTerritory.armies.toString());
			if (defensePower > attackPower) {
				target.armies = defensePower - attackPower;
				target.armyText.setText(target.armies.toString());
			}
		}
		LK.getSound('attack').play();
		updateTerritoryCount();
		endPlayerTurn();
	}
}
// Recruit armies action
function recruitArmies() {
	if (!selectedTerritory) return;
	selectedTerritory.addArmies(2);
	endPlayerTurn();
}
// End player turn
function endPlayerTurn() {
	if (selectedTerritory) {
		selectedTerritory.selected = false;
		selectedTerritory.updateGraphics();
		selectedTerritory = null;
	}
	currentTurn = 2;
	updateInstructions();
	// AI turn after short delay
	LK.setTimeout(function () {
		aiTurn();
	}, 1000);
}
// Simple AI turn
function aiTurn() {
	var enemyTerrs = [];
	for (var i = 0; i < territories.length; i++) {
		if (territories[i].owner === 2) {
			enemyTerrs.push(territories[i]);
		}
	}
	if (enemyTerrs.length > 0) {
		var aiTerritory = enemyTerrs[Math.floor(Math.random() * enemyTerrs.length)];
		var neighbors = getNeighbors(aiTerritory);
		var playerTargets = [];
		for (var j = 0; j < neighbors.length; j++) {
			if (neighbors[j].owner === 1) {
				playerTargets.push(neighbors[j]);
			}
		}
		if (playerTargets.length > 0 && aiTerritory.armies > 1) {
			// AI attacks
			var target = playerTargets[Math.floor(Math.random() * playerTargets.length)];
			var attackPower = aiTerritory.armies - 1;
			var defensePower = target.armies;
			if (attackPower > defensePower) {
				target.setOwner(2, attackPower - defensePower);
				aiTerritory.armies = 1;
				aiTerritory.armyText.setText(aiTerritory.armies.toString());
			} else {
				aiTerritory.armies = 1;
				aiTerritory.armyText.setText(aiTerritory.armies.toString());
				if (defensePower > attackPower) {
					target.armies = defensePower - attackPower;
					target.armyText.setText(target.armies.toString());
				}
			}
		} else {
			// AI recruits
			aiTerritory.addArmies(1);
		}
	}
	updateTerritoryCount();
	currentTurn = 1;
	updateInstructions();
}
// Create UI elements
function createUI() {
	// Turn indicator
	turnText = new Text2("Your Turn", {
		size: 40,
		fill: 0xFFFFFF
	});
	turnText.anchor.set(0.5, 0);
	LK.gui.top.addChild(turnText);
	turnText.y = 100;
	// Score display
	scoreText = new Text2("Player: 0 | Enemy: 0", {
		size: 32,
		fill: 0xFFFFFF
	});
	scoreText.anchor.set(0.5, 0);
	LK.gui.top.addChild(scoreText);
	scoreText.y = 150;
	// Player army count display
	playerArmyText = new Text2("Total Armies: 0", {
		size: 32,
		fill: 0x7ed321
	});
	playerArmyText.anchor.set(0.5, 0);
	LK.gui.top.addChild(playerArmyText);
	playerArmyText.y = 190;
	// Instructions
	instructionText = new Text2("Select your territory, then choose an action", {
		size: 28,
		fill: 0xFFFFFF
	});
	instructionText.anchor.set(0.5, 1);
	LK.gui.bottom.addChild(instructionText);
	instructionText.y = -200;
	// Action buttons
	var attackBtn = new ActionButton();
	attackBtn.init("Attack", attackTerritory, 2048 / 2 - 100, 2732 - 120);
	actionButtons.push(attackBtn);
	game.addChild(attackBtn);
	var recruitBtn = new ActionButton();
	recruitBtn.init("Recruit", recruitArmies, 2048 / 2 + 100, 2732 - 120);
	actionButtons.push(recruitBtn);
	game.addChild(recruitBtn);
}
// Update instruction text
function updateInstructions() {
	if (currentTurn === 1) {
		turnText.setText("Your Turn");
		if (selectedTerritory) {
			instructionText.setText("Territory selected - Choose Attack or Recruit");
		} else {
			instructionText.setText("Select your territory (green), then choose an action");
		}
	} else {
		turnText.setText("Enemy Turn");
		instructionText.setText("Enemy is thinking...");
	}
}
// Initialize the game
createMap();
createUI();
updateInstructions();

===================================================================
--- original.js
+++ change.js
@@ -108,9 +108,9 @@
 var playerTerritories = 0;
 var enemyTerritories = 0;
 var totalTerritories = 0;
 // UI elements
-var turnText, scoreText, instructionText;
+var turnText, scoreText, instructionText, playerArmyText;
 var actionButtons = [];
 // Game settings
 var mapWidth = 6;
 var mapHeight = 8;
@@ -145,16 +145,26 @@
 // Update territory ownership counts
 function updateTerritoryCount() {
 	playerTerritories = 0;
 	enemyTerritories = 0;
+	var playerTotalArmies = 0;
 	totalTerritories = territories.length;
 	for (var i = 0; i < territories.length; i++) {
-		if (territories[i].owner === 1) playerTerritories++;else if (territories[i].owner === 2) enemyTerritories++;
+		if (territories[i].owner === 1) {
+			playerTerritories++;
+			playerTotalArmies += territories[i].armies;
+		} else if (territories[i].owner === 2) {
+			enemyTerritories++;
+		}
 	}
 	// Update score display
 	if (scoreText) {
 		scoreText.setText("Player: " + playerTerritories + " | Enemy: " + enemyTerritories);
 	}
+	// Update player army display
+	if (playerArmyText) {
+		playerArmyText.setText("Total Armies: " + playerTotalArmies);
+	}
 	// Check victory conditions
 	if (playerTerritories >= Math.floor(totalTerritories * 0.75)) {
 		gameState = 'victory';
 		LK.showYouWin();
@@ -332,8 +342,16 @@
 	});
 	scoreText.anchor.set(0.5, 0);
 	LK.gui.top.addChild(scoreText);
 	scoreText.y = 150;
+	// Player army count display
+	playerArmyText = new Text2("Total Armies: 0", {
+		size: 32,
+		fill: 0x7ed321
+	});
+	playerArmyText.anchor.set(0.5, 0);
+	LK.gui.top.addChild(playerArmyText);
+	playerArmyText.y = 190;
 	// Instructions
 	instructionText = new Text2("Select your territory, then choose an action", {
 		size: 28,
 		fill: 0xFFFFFF