===================================================================
--- original.js
+++ change.js
@@ -1,6 +1,455 @@
-/****
+/**** 
+* Plugins
+****/ 
+var tween = LK.import("@upit/tween.v1");
+
+/**** 
+* Classes
+****/ 
+var Plant = Container.expand(function (type) {
+	var self = Container.call(this);
+	self.type = type;
+	self.health = 100;
+	self.maxHealth = 100;
+	self.shootTimer = 0;
+	self.shootCooldown = 60; // 1 second at 60fps
+	if (type === 'peashooter') {
+		self.graphics = self.attachAsset('peashooter', {
+			anchorX: 0.5,
+			anchorY: 0.5
+		});
+		self.damage = 25;
+		self.range = 400;
+	} else if (type === 'wallnut') {
+		self.graphics = self.attachAsset('wallnut', {
+			anchorX: 0.5,
+			anchorY: 0.5
+		});
+		self.health = 300;
+		self.maxHealth = 300;
+		self.damage = 0;
+	} else if (type === 'cherrybomb') {
+		self.graphics = self.attachAsset('cherrybomb', {
+			anchorX: 0.5,
+			anchorY: 0.5
+		});
+		self.damage = 200;
+		self.explosionRadius = 150;
+		self.fuseTimer = 180; // 3 seconds
+	}
+	self.update = function () {
+		if (self.type === 'peashooter') {
+			self.shootTimer++;
+			if (self.shootTimer >= self.shootCooldown) {
+				var target = self.findNearestZombie();
+				if (target && self.getDistanceTo(target) <= self.range) {
+					self.shoot(target);
+					self.shootTimer = 0;
+				}
+			}
+		} else if (self.type === 'cherrybomb') {
+			self.fuseTimer--;
+			if (self.fuseTimer <= 0) {
+				self.explode();
+			}
+		}
+	};
+	self.findNearestZombie = function () {
+		var nearest = null;
+		var minDistance = Infinity;
+		for (var i = 0; i < zombies.length; i++) {
+			var zombie = zombies[i];
+			var distance = self.getDistanceTo(zombie);
+			if (distance < minDistance) {
+				minDistance = distance;
+				nearest = zombie;
+			}
+		}
+		return nearest;
+	};
+	self.getDistanceTo = function (target) {
+		var dx = target.x - self.x;
+		var dy = target.y - self.y;
+		return Math.sqrt(dx * dx + dy * dy);
+	};
+	self.shoot = function (target) {
+		var projectile = new Projectile(self.x, self.y, target, self.damage);
+		projectiles.push(projectile);
+		game.addChild(projectile);
+		LK.getSound('shoot').play();
+	};
+	self.explode = function () {
+		for (var i = zombies.length - 1; i >= 0; i--) {
+			var zombie = zombies[i];
+			if (self.getDistanceTo(zombie) <= self.explosionRadius) {
+				zombie.takeDamage(self.damage);
+			}
+		}
+		LK.getSound('explosion').play();
+		self.destroy();
+		plants.splice(plants.indexOf(self), 1);
+	};
+	self.takeDamage = function (damage) {
+		self.health -= damage;
+		if (self.health <= 0) {
+			self.destroy();
+			plants.splice(plants.indexOf(self), 1);
+		}
+	};
+	return self;
+});
+var PlantButton = Container.expand(function (plantType, index) {
+	var self = Container.call(this);
+	self.plantType = plantType;
+	self.button = self.attachAsset('plantButton', {
+		anchorX: 0.5,
+		anchorY: 0.5
+	});
+	var icon;
+	if (plantType === 'peashooter') {
+		icon = self.attachAsset('peashooter', {
+			anchorX: 0.5,
+			anchorY: 0.5,
+			scaleX: 0.6,
+			scaleY: 0.6
+		});
+	} else if (plantType === 'wallnut') {
+		icon = self.attachAsset('wallnut', {
+			anchorX: 0.5,
+			anchorY: 0.5,
+			scaleX: 0.6,
+			scaleY: 0.6
+		});
+	} else if (plantType === 'cherrybomb') {
+		icon = self.attachAsset('cherrybomb', {
+			anchorX: 0.5,
+			anchorY: 0.5,
+			scaleX: 0.6,
+			scaleY: 0.6
+		});
+	}
+	self.x = 150 + index * 140;
+	self.y = 80;
+	self.down = function (x, y, obj) {
+		selectedPlantType = self.plantType;
+		// Visual feedback
+		for (var i = 0; i < plantButtons.length; i++) {
+			plantButtons[i].button.tint = 0xFFFFFF;
+		}
+		self.button.tint = 0x00FF00;
+	};
+	return self;
+});
+var Projectile = Container.expand(function (startX, startY, target, damage) {
+	var self = Container.call(this);
+	self.graphics = self.attachAsset('pea', {
+		anchorX: 0.5,
+		anchorY: 0.5
+	});
+	self.x = startX;
+	self.y = startY;
+	self.target = target;
+	self.damage = damage;
+	self.speed = 8;
+	// Calculate direction
+	var dx = target.x - startX;
+	var dy = target.y - startY;
+	var distance = Math.sqrt(dx * dx + dy * dy);
+	self.dirX = dx / distance;
+	self.dirY = dy / distance;
+	self.update = function () {
+		self.x += self.dirX * self.speed;
+		self.y += self.dirY * self.speed;
+		// Check collision with target or any zombie
+		for (var i = 0; i < zombies.length; i++) {
+			var zombie = zombies[i];
+			if (Math.abs(zombie.x - self.x) < 30 && Math.abs(zombie.y - self.y) < 30) {
+				zombie.takeDamage(self.damage);
+				self.destroy();
+				projectiles.splice(projectiles.indexOf(self), 1);
+				return;
+			}
+		}
+		// Remove if off-screen
+		if (self.x > 2100 || self.x < -50 || self.y > 2800 || self.y < -50) {
+			self.destroy();
+			projectiles.splice(projectiles.indexOf(self), 1);
+		}
+	};
+	return self;
+});
+var Zombie = Container.expand(function (type, lane) {
+	var self = Container.call(this);
+	self.type = type;
+	self.lane = lane;
+	self.speed = 1;
+	self.health = 100;
+	self.maxHealth = 100;
+	self.damage = 25;
+	self.attackCooldown = 60;
+	self.attackTimer = 0;
+	self.isAttacking = false;
+	if (type === 'basic') {
+		self.graphics = self.attachAsset('basicZombie', {
+			anchorX: 0.5,
+			anchorY: 0.5
+		});
+		self.speed = 0.5;
+		self.health = 100;
+		self.maxHealth = 100;
+	} else if (type === 'fast') {
+		self.graphics = self.attachAsset('fastZombie', {
+			anchorX: 0.5,
+			anchorY: 0.5
+		});
+		self.speed = 1.2;
+		self.health = 60;
+		self.maxHealth = 60;
+	} else if (type === 'tank') {
+		self.graphics = self.attachAsset('tankZombie', {
+			anchorX: 0.5,
+			anchorY: 0.5
+		});
+		self.speed = 0.3;
+		self.health = 300;
+		self.maxHealth = 300;
+		self.damage = 50;
+	}
+	self.x = 2048 + 50; // Start off-screen right
+	self.y = 200 + lane * 120; // Position in lane
+	self.update = function () {
+		if (!self.isAttacking) {
+			self.x -= self.speed;
+			// Check for plant collision
+			var plantInWay = self.findPlantInWay();
+			if (plantInWay) {
+				self.isAttacking = true;
+				self.attackTimer = 0;
+			}
+		} else {
+			self.attackTimer++;
+			if (self.attackTimer >= self.attackCooldown) {
+				var plantInWay = self.findPlantInWay();
+				if (plantInWay) {
+					plantInWay.takeDamage(self.damage);
+					self.attackTimer = 0;
+				} else {
+					self.isAttacking = false;
+				}
+			}
+		}
+		// Check if reached end
+		if (self.x < -50) {
+			zombiesReachedEnd++;
+			self.destroy();
+			zombies.splice(zombies.indexOf(self), 1);
+		}
+	};
+	self.findPlantInWay = function () {
+		for (var i = 0; i < plants.length; i++) {
+			var plant = plants[i];
+			if (Math.abs(plant.x - self.x) < 50 && Math.abs(plant.y - self.y) < 50) {
+				return plant;
+			}
+		}
+		return null;
+	};
+	self.takeDamage = function (damage) {
+		self.health -= damage;
+		LK.getSound('zombieHit').play();
+		// Visual feedback
+		tween(self.graphics, {
+			tint: 0xFF0000
+		}, {
+			duration: 200
+		});
+		tween(self.graphics, {
+			tint: 0xFFFFFF
+		}, {
+			duration: 200
+		});
+		if (self.health <= 0) {
+			LK.setScore(LK.getScore() + 10);
+			scoreText.setText('Score: ' + LK.getScore());
+			self.destroy();
+			zombies.splice(zombies.indexOf(self), 1);
+		}
+	};
+	return self;
+});
+
+/**** 
 * Initialize Game
-****/
+****/ 
 var game = new LK.Game({
-	backgroundColor: 0x000000
-});
\ No newline at end of file
+	backgroundColor: 0x2E7D32
+});
+
+/**** 
+* Game Code
+****/ 
+// Sounds
+// Grid and UI
+// Projectiles
+// Zombies
+// Plants
+// Game variables
+var plants = [];
+var zombies = [];
+var projectiles = [];
+var plantButtons = [];
+var gridCells = [];
+var selectedPlantType = null;
+var waveNumber = 1;
+var zombiesInWave = 5;
+var zombieSpawnTimer = 0;
+var zombieSpawnInterval = 120; // 2 seconds
+var zombiesSpawned = 0;
+var zombiesReachedEnd = 0;
+var maxZombiesAllowed = 5;
+var gameState = 'playing'; // 'playing', 'gameOver', 'victory'
+// UI Elements
+var scoreText = new Text2('Score: 0', {
+	size: 40,
+	fill: '#FFFFFF'
+});
+scoreText.anchor.set(0, 0);
+scoreText.x = 150;
+scoreText.y = 20;
+LK.gui.topLeft.addChild(scoreText);
+var waveText = new Text2('Wave: 1', {
+	size: 40,
+	fill: '#FFFFFF'
+});
+waveText.anchor.set(0.5, 0);
+waveText.x = 1024;
+waveText.y = 20;
+LK.gui.top.addChild(waveText);
+var livesText = new Text2('Lives: 5', {
+	size: 40,
+	fill: '#FF0000'
+});
+livesText.anchor.set(1, 0);
+livesText.x = 1900;
+livesText.y = 20;
+LK.gui.topRight.addChild(livesText);
+// Create plant selection buttons
+var plantTypes = ['peashooter', 'wallnut', 'cherrybomb'];
+for (var i = 0; i < plantTypes.length; i++) {
+	var button = new PlantButton(plantTypes[i], i);
+	plantButtons.push(button);
+	game.addChild(button);
+}
+// Create grid
+var gridStartX = 200;
+var gridStartY = 180;
+var gridRows = 5;
+var gridCols = 8;
+for (var row = 0; row < gridRows; row++) {
+	gridCells[row] = [];
+	for (var col = 0; col < gridCols; col++) {
+		var cell = LK.getAsset('gridCell', {
+			anchorX: 0.5,
+			anchorY: 0.5,
+			alpha: 0.3
+		});
+		cell.x = gridStartX + col * 120;
+		cell.y = gridStartY + row * 120;
+		cell.row = row;
+		cell.col = col;
+		cell.occupied = false;
+		gridCells[row][col] = cell;
+		game.addChild(cell);
+	}
+}
+// Game event handlers
+game.down = function (x, y, obj) {
+	if (!selectedPlantType) return;
+	// Check if clicking on grid
+	for (var row = 0; row < gridRows; row++) {
+		for (var col = 0; col < gridCols; col++) {
+			var cell = gridCells[row][col];
+			if (Math.abs(x - cell.x) < 50 && Math.abs(y - cell.y) < 50 && !cell.occupied) {
+				var plant = new Plant(selectedPlantType);
+				plant.x = cell.x;
+				plant.y = cell.y;
+				plants.push(plant);
+				game.addChild(plant);
+				cell.occupied = true;
+				LK.getSound('plantPlace').play();
+				// Reset selection
+				selectedPlantType = null;
+				for (var i = 0; i < plantButtons.length; i++) {
+					plantButtons[i].button.tint = 0xFFFFFF;
+				}
+				return;
+			}
+		}
+	}
+};
+function spawnZombie() {
+	if (zombiesSpawned >= zombiesInWave) return;
+	var lane = Math.floor(Math.random() * gridRows);
+	var zombieTypes = ['basic', 'fast', 'tank'];
+	var type = zombieTypes[Math.floor(Math.random() * zombieTypes.length)];
+	// Increase difficulty with wave number
+	if (waveNumber > 3) {
+		type = zombieTypes[Math.floor(Math.random() * zombieTypes.length)];
+	} else if (waveNumber > 1) {
+		type = Math.random() < 0.7 ? 'basic' : 'fast';
+	} else {
+		type = 'basic';
+	}
+	var zombie = new Zombie(type, lane);
+	zombies.push(zombie);
+	game.addChild(zombie);
+	zombiesSpawned++;
+}
+function checkWaveComplete() {
+	if (zombiesSpawned >= zombiesInWave && zombies.length === 0) {
+		// Wave complete
+		waveNumber++;
+		zombiesInWave += 2;
+		zombiesSpawned = 0;
+		waveText.setText('Wave: ' + waveNumber);
+		if (waveNumber > 10) {
+			gameState = 'victory';
+			LK.showYouWin();
+		}
+	}
+}
+function checkGameOver() {
+	if (zombiesReachedEnd >= maxZombiesAllowed) {
+		gameState = 'gameOver';
+		LK.showGameOver();
+	}
+	livesText.setText('Lives: ' + (maxZombiesAllowed - zombiesReachedEnd));
+}
+// Main game loop
+game.update = function () {
+	if (gameState !== 'playing') return;
+	// Spawn zombies
+	zombieSpawnTimer++;
+	if (zombieSpawnTimer >= zombieSpawnInterval) {
+		spawnZombie();
+		zombieSpawnTimer = 0;
+	}
+	// Update all objects
+	for (var i = plants.length - 1; i >= 0; i--) {
+		if (plants[i].update) {
+			plants[i].update();
+		}
+	}
+	for (var i = zombies.length - 1; i >= 0; i--) {
+		if (zombies[i].update) {
+			zombies[i].update();
+		}
+	}
+	for (var i = projectiles.length - 1; i >= 0; i--) {
+		if (projectiles[i].update) {
+			projectiles[i].update();
+		}
+	}
+	checkWaveComplete();
+	checkGameOver();
+};
\ No newline at end of file