/**** 
* Plugins
****/ 
var tween = LK.import("@upit/tween.v1");

/**** 
* Classes
****/ 
var Lawnmower = Container.expand(function () {
	var self = Container.call(this);
	var lawnmowerGraphics = self.attachAsset('lawnmower', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.speed = 0; // Starts stationary
	self.activated = false;
	self.update = function () {
		if (self.activated) {
			self.x += 12; // Move fast when activated
			if (self.x > 2200) {
				self.destroy();
			}
		}
	};
	return self;
});
var Pea = Container.expand(function () {
	var self = Container.call(this);
	var peaGraphics = self.attachAsset('pea', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.speed = 8;
	self.damage = 20;
	self.update = function () {
		self.x += self.speed;
		if (self.x > 2200) {
			self.destroy();
		}
	};
	return self;
});
var Plant = Container.expand(function (type) {
	var self = Container.call(this);
	self.type = type;
	self.health = 100;
	self.maxHealth = 100;
	self.shootCooldown = 0;
	self.sunTimer = 0;
	var plantGraphics;
	if (type === 'sunflower') {
		plantGraphics = self.attachAsset('sunflower', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.sunGenerateRate = 300; // ticks between sun generation
	} else if (type === 'peashooter') {
		plantGraphics = self.attachAsset('peashooter', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.shootRate = 90; // ticks between shots
	} else if (type === 'wallnut') {
		plantGraphics = self.attachAsset('wallnut', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.health = 300;
		self.maxHealth = 300;
	} else if (type === 'potatomine') {
		plantGraphics = self.attachAsset('potatomine', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.health = 50;
		self.maxHealth = 50;
		self.exploded = false;
	} else if (type === 'carnivorousplant') {
		plantGraphics = self.attachAsset('carnivorousplant', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.health = 200;
		self.maxHealth = 200;
		self.attackCooldown = 0;
		self.attackRange = 100;
	} else if (type === 'doomshroom') {
		plantGraphics = self.attachAsset('doomshroom', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.health = 100;
		self.maxHealth = 100;
		self.exploded = false;
		self.armingTime = 900; // 15 seconds to arm
		self.currentArmingTime = 0;
	} else if (type === 'repeater') {
		plantGraphics = self.attachAsset('repeater', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.shootRate = 90; // Same as peashooter
	} else if (type === 'threepeater') {
		plantGraphics = self.attachAsset('threepeater', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.shootRate = 120; // Slightly slower due to triple shot
	}
	self.takeDamage = function (damage) {
		self.health -= damage;
		if (self.health <= 0) {
			self.destroy();
			return true;
		}
		return false;
	};
	self.update = function () {
		if (self.type === 'sunflower') {
			self.sunTimer++;
			if (self.sunTimer >= self.sunGenerateRate) {
				self.sunTimer = 0;
				var newSun = new Sun();
				newSun.x = self.x + Math.random() * 40 - 20;
				newSun.y = self.y + Math.random() * 40 - 20;
				game.addChild(newSun);
				suns.push(newSun);
			}
		} else if (self.type === 'peashooter') {
			self.shootCooldown++;
			if (self.shootCooldown >= self.shootRate) {
				// Check if there's a zombie in this row
				var myRow = Math.floor((self.y - 500) / 140);
				var hasZombieInRow = false;
				for (var i = 0; i < zombies.length; i++) {
					var zombie = zombies[i];
					var zombieRow = Math.floor((zombie.y - 500) / 140);
					if (zombieRow === myRow && zombie.x > self.x) {
						hasZombieInRow = true;
						break;
					}
				}
				if (hasZombieInRow) {
					self.shootCooldown = 0;
					var pea = new Pea();
					pea.x = self.x + 60;
					pea.y = self.y;
					game.addChild(pea);
					peas.push(pea);
					LK.getSound('shoot').play();
				}
			}
		} else if (self.type === 'repeater') {
			self.shootCooldown++;
			if (self.shootCooldown >= self.shootRate) {
				// Check if there's a zombie in this row
				var myRow = Math.floor((self.y - 500) / 140);
				var hasZombieInRow = false;
				for (var i = 0; i < zombies.length; i++) {
					var zombie = zombies[i];
					var zombieRow = Math.floor((zombie.y - 500) / 140);
					if (zombieRow === myRow && zombie.x > self.x) {
						hasZombieInRow = true;
						break;
					}
				}
				if (hasZombieInRow) {
					self.shootCooldown = 0;
					// Shoot two peas with slight delay
					var pea1 = new Pea();
					pea1.x = self.x + 60;
					pea1.y = self.y;
					game.addChild(pea1);
					peas.push(pea1);
					// Second pea with slight delay
					LK.setTimeout(function () {
						var pea2 = new Pea();
						pea2.x = self.x + 60;
						pea2.y = self.y;
						game.addChild(pea2);
						peas.push(pea2);
					}, 150);
					LK.getSound('shoot').play();
				}
			}
		} else if (self.type === 'threepeater') {
			self.shootCooldown++;
			if (self.shootCooldown >= self.shootRate) {
				// Check if there's a zombie in any of the three rows (current, above, below)
				var myRow = Math.floor((self.y - 500) / 140);
				var hasZombieInAnyRow = false;
				for (var i = 0; i < zombies.length; i++) {
					var zombie = zombies[i];
					var zombieRow = Math.floor((zombie.y - 500) / 140);
					if ((zombieRow === myRow || zombieRow === myRow - 1 || zombieRow === myRow + 1) && zombie.x > self.x) {
						hasZombieInAnyRow = true;
						break;
					}
				}
				if (hasZombieInAnyRow) {
					self.shootCooldown = 0;
					// Shoot three peas - center, above, below
					var centerPea = new Pea();
					centerPea.x = self.x + 60;
					centerPea.y = self.y;
					game.addChild(centerPea);
					peas.push(centerPea);
					// Above pea (if row exists)
					if (myRow > 0) {
						var abovePea = new Pea();
						abovePea.x = self.x + 60;
						abovePea.y = self.y - 140;
						game.addChild(abovePea);
						peas.push(abovePea);
					}
					// Below pea (if row exists)
					if (myRow < gridRows - 1) {
						var belowPea = new Pea();
						belowPea.x = self.x + 60;
						belowPea.y = self.y + 140;
						game.addChild(belowPea);
						peas.push(belowPea);
					}
					LK.getSound('shoot').play();
				}
			}
		} else if (self.type === 'potatomine' && !self.exploded) {
			// Check for zombies stepping on the mine
			var myRow = Math.floor((self.y - 500) / 140);
			for (var i = 0; i < zombies.length; i++) {
				var zombie = zombies[i];
				var zombieRow = Math.floor((zombie.y - 500) / 140);
				if (zombieRow === myRow && Math.abs(self.x - zombie.x) < 80) {
					// Explode!
					self.exploded = true;
					LK.getSound('explode').play();
					// Create explosion visual effect
					tween(self, {
						scaleX: 3,
						scaleY: 3,
						alpha: 0
					}, {
						duration: 500,
						easing: tween.easeOut,
						onFinish: function onFinish() {
							for (var k = plants.length - 1; k >= 0; k--) {
								if (plants[k] === self) {
									plants.splice(k, 1);
									self.destroy();
									break;
								}
							}
						}
					});
					LK.effects.flashObject(self, 0xFF4500, 500);
					// Damage all zombies in the area
					for (var j = zombies.length - 1; j >= 0; j--) {
						var targetZombie = zombies[j];
						var distance = Math.sqrt(Math.pow(self.x - targetZombie.x, 2) + Math.pow(self.y - targetZombie.y, 2));
						if (distance < 120) {
							if (targetZombie.takeDamage(200)) {
								targetZombie.destroy();
								zombies.splice(j, 1);
							}
						}
					}
					break;
				}
			}
		} else if (self.type === 'carnivorousplant') {
			self.attackCooldown++;
			if (self.attackCooldown >= 180) {
				// Attack every 3 seconds
				var myRow = Math.floor((self.y - 500) / 140);
				var closestZombie = null;
				var closestDistance = Infinity;
				for (var i = 0; i < zombies.length; i++) {
					var zombie = zombies[i];
					var zombieRow = Math.floor((zombie.y - 500) / 140);
					if (zombieRow === myRow) {
						var distance = Math.abs(self.x - zombie.x);
						if (distance < self.attackRange && distance < closestDistance) {
							closestDistance = distance;
							closestZombie = zombie;
						}
					}
				}
				if (closestZombie) {
					self.attackCooldown = 0;
					// Attack animation
					tween(self, {
						scaleX: 1.5,
						scaleY: 1.5
					}, {
						duration: 200,
						easing: tween.easeOut,
						onFinish: function onFinish() {
							tween(self, {
								scaleX: 1,
								scaleY: 1
							}, {
								duration: 200,
								easing: tween.easeIn
							});
						}
					});
					if (closestZombie.takeDamage(100)) {
						for (var j = zombies.length - 1; j >= 0; j--) {
							if (zombies[j] === closestZombie) {
								closestZombie.destroy();
								zombies.splice(j, 1);
								break;
							}
						}
					}
				}
			}
		} else if (self.type === 'doomshroom' && !self.exploded) {
			self.currentArmingTime++;
			if (self.currentArmingTime >= self.armingTime) {
				// Doom shroom is now armed and explodes
				self.exploded = true;
				LK.getSound('explode').play();
				// Create massive explosion visual effect
				tween(self, {
					scaleX: 8,
					scaleY: 8,
					alpha: 0
				}, {
					duration: 1000,
					easing: tween.easeOut,
					onFinish: function onFinish() {
						for (var k = plants.length - 1; k >= 0; k--) {
							if (plants[k] === self) {
								plants.splice(k, 1);
								self.destroy();
								break;
							}
						}
					}
				});
				LK.effects.flashScreen(0x4B0082, 1000);
				// Damage all zombies on screen
				for (var j = zombies.length - 1; j >= 0; j--) {
					var targetZombie = zombies[j];
					if (targetZombie.takeDamage(300)) {
						targetZombie.destroy();
						zombies.splice(j, 1);
					}
				}
			}
		}
	};
	return self;
});
var Sun = Container.expand(function () {
	var self = Container.call(this);
	var sunGraphics = self.attachAsset('sun', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.value = 25;
	self.collected = false;
	self.down = function (x, y, obj) {
		if (!self.collected) {
			self.collected = true;
			sunPoints += self.value;
			updateSunDisplay();
			LK.getSound('collect').play();
			self.destroy();
		}
	};
	return self;
});
var Zombie = Container.expand(function (type) {
	var self = Container.call(this);
	self.type = type || 'normal';
	self.health = 100;
	self.maxHealth = 100;
	self.speed = -1;
	self.attackDamage = 25;
	self.attackCooldown = 0;
	self.isAttacking = false;
	var zombieGraphics;
	if (self.type === 'fast') {
		zombieGraphics = self.attachAsset('fastZombie', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.speed = -2;
		self.health = 60;
		self.maxHealth = 60;
	} else if (self.type === 'doctor') {
		zombieGraphics = self.attachAsset('zombieDoctor', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.speed = -0.8;
		self.health = 150;
		self.maxHealth = 150;
		self.healCooldown = 0;
	} else if (self.type === 'snailBucket') {
		zombieGraphics = self.attachAsset('snailBucket', {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.speed = -0.5;
		self.health = 250;
		self.maxHealth = 250;
	} else {
		zombieGraphics = self.attachAsset('zombie', {
			anchorX: 0.5,
			anchorY: 0.5
		});
	}
	self.takeDamage = function (damage) {
		self.health -= damage;
		if (self.health <= 0) {
			return true;
		}
		return false;
	};
	self.update = function () {
		// Doctor zombie healing behavior
		if (self.type === 'doctor') {
			self.healCooldown++;
			if (self.healCooldown >= 300) {
				// Heal every 5 seconds
				self.healCooldown = 0;
				// Heal nearby zombies
				for (var k = 0; k < zombies.length; k++) {
					var nearbyZombie = zombies[k];
					if (nearbyZombie !== self) {
						var distance = Math.sqrt(Math.pow(self.x - nearbyZombie.x, 2) + Math.pow(self.y - nearbyZombie.y, 2));
						if (distance < 150) {
							nearbyZombie.health = Math.min(nearbyZombie.health + 30, nearbyZombie.maxHealth);
							tween(nearbyZombie, {
								tint: 0x00ff00
							}, {
								duration: 300,
								easing: tween.easeOut,
								onFinish: function onFinish() {
									tween(nearbyZombie, {
										tint: 0xffffff
									}, {
										duration: 300,
										easing: tween.easeIn
									});
								}
							});
						}
					}
				}
			}
		}
		if (!self.isAttacking) {
			self.x += self.speed;
			// Check if zombie reached the left side
			if (self.x < 100) {
				gameOver();
				return;
			}
			// Check for plants to attack
			var myRow = Math.floor((self.y - 500) / 140);
			for (var i = 0; i < plants.length; i++) {
				var plant = plants[i];
				var plantRow = Math.floor((plant.y - 500) / 140);
				if (plantRow === myRow && Math.abs(self.x - plant.x) < 80) {
					self.isAttacking = true;
					break;
				}
			}
		} else {
			// Attack mode
			self.attackCooldown++;
			if (self.attackCooldown >= 60) {
				// Attack every second
				self.attackCooldown = 0;
				// Find plant to attack
				var myRow = Math.floor((self.y - 500) / 140);
				var targetPlant = null;
				for (var i = 0; i < plants.length; i++) {
					var plant = plants[i];
					var plantRow = Math.floor((plant.y - 500) / 140);
					if (plantRow === myRow && Math.abs(self.x - plant.x) < 80) {
						targetPlant = plant;
						break;
					}
				}
				if (targetPlant) {
					if (targetPlant.takeDamage(self.attackDamage)) {
						// Plant destroyed, remove from array
						for (var j = plants.length - 1; j >= 0; j--) {
							if (plants[j] === targetPlant) {
								plants.splice(j, 1);
								break;
							}
						}
						self.isAttacking = false;
					}
				} else {
					self.isAttacking = false;
				}
			}
		}
	};
	return self;
});

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

/**** 
* Game Code
****/ 
// Game variables
var sunPoints = 150;
var currentWave = 1;
var maxWaves = 5;
var waveSpawnTimer = 0;
var zombiesInWave = 0;
var maxZombiesInWave = 5;
var selectedPlantType = null;
var shovelSelected = false;
var gameState = 'playing'; // 'playing', 'won', 'lost'
// Seed recharge system - 3 seconds = 180 ticks at 60 FPS
var seedRechargeTime = 180;
var seedCooldowns = {
	sunflower: 0,
	peashooter: 0,
	wallnut: 0,
	potatomine: 0,
	carnivorousplant: 0,
	doomshroom: 0,
	repeater: 0,
	threepeater: 0
};
// Game arrays
var plants = [];
var zombies = [];
var peas = [];
var suns = [];
var lawnmowers = [];
// Grid setup
var gridStartX = 300;
var gridStartY = 500;
var gridCols = 12;
var gridRows = 12;
var cellSize = 140;
// Plant costs
var plantCosts = {
	sunflower: 50,
	peashooter: 100,
	wallnut: 50,
	potatomine: 25,
	carnivorousplant: 150,
	doomshroom: 125,
	repeater: 200,
	threepeater: 325
};
// UI Elements
var sunDisplay = new Text2('Sun: ' + sunPoints, {
	size: 60,
	fill: 0xFFFF00
});
sunDisplay.anchor.set(0, 0);
LK.gui.topLeft.addChild(sunDisplay);
sunDisplay.x = 120;
sunDisplay.y = 20;
var waveDisplay = new Text2('Wave: ' + currentWave + '/' + maxWaves, {
	size: 50,
	fill: 0xFFFFFF
});
waveDisplay.anchor.set(0.5, 0);
LK.gui.top.addChild(waveDisplay);
waveDisplay.y = 20;
// Plant selection buttons with unique sprites
var sunflowerButton = LK.getAsset('sunflower', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 400,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(sunflowerButton);
var peashooterButton = LK.getAsset('peashooter', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 520,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(peashooterButton);
var wallnutButton = LK.getAsset('wallnut', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 640,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(wallnutButton);
var potatomineButton = LK.getAsset('potatomine', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 760,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(potatomineButton);
var carnivorousplantButton = LK.getAsset('carnivorousplant', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 880,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(carnivorousplantButton);
var doomshroomButton = LK.getAsset('doomshroom', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 1000,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(doomshroomButton);
var repeaterButton = LK.getAsset('repeater', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 1120,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(repeaterButton);
var threepeaterButton = LK.getAsset('threepeater', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 1240,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(threepeaterButton);
var shovelButton = LK.getAsset('shovel', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 100,
	y: 1360,
	scaleX: 0.8,
	scaleY: 0.8
});
game.addChild(shovelButton);
// Plant cost labels
var sunflowerCost = new Text2('50', {
	size: 30,
	fill: 0xFFFFFF
});
sunflowerCost.anchor.set(0.5, 0.5);
sunflowerCost.x = 100;
sunflowerCost.y = 450;
game.addChild(sunflowerCost);
var peashooterCost = new Text2('100', {
	size: 30,
	fill: 0xFFFFFF
});
peashooterCost.anchor.set(0.5, 0.5);
peashooterCost.x = 100;
peashooterCost.y = 570;
game.addChild(peashooterCost);
var wallnutCost = new Text2('50', {
	size: 30,
	fill: 0xFFFFFF
});
wallnutCost.anchor.set(0.5, 0.5);
wallnutCost.x = 100;
wallnutCost.y = 690;
game.addChild(wallnutCost);
var potatomineeCost = new Text2('25', {
	size: 30,
	fill: 0xFFFFFF
});
potatomineeCost.anchor.set(0.5, 0.5);
potatomineeCost.x = 100;
potatomineeCost.y = 810;
game.addChild(potatomineeCost);
var carnivorousplantCost = new Text2('150', {
	size: 30,
	fill: 0xFFFFFF
});
carnivorousplantCost.anchor.set(0.5, 0.5);
carnivorousplantCost.x = 100;
carnivorousplantCost.y = 930;
game.addChild(carnivorousplantCost);
var doomshroomCost = new Text2('125', {
	size: 30,
	fill: 0xFFFFFF
});
doomshroomCost.anchor.set(0.5, 0.5);
doomshroomCost.x = 100;
doomshroomCost.y = 1050;
game.addChild(doomshroomCost);
var repeaterCost = new Text2('200', {
	size: 30,
	fill: 0xFFFFFF
});
repeaterCost.anchor.set(0.5, 0.5);
repeaterCost.x = 100;
repeaterCost.y = 1170;
game.addChild(repeaterCost);
var threepeaterCost = new Text2('325', {
	size: 30,
	fill: 0xFFFFFF
});
threepeaterCost.anchor.set(0.5, 0.5);
threepeaterCost.x = 100;
threepeaterCost.y = 1290;
game.addChild(threepeaterCost);
// Draw grid
for (var row = 0; row < gridRows; row++) {
	for (var col = 0; col < gridCols; col++) {
		var gridCell = LK.getAsset('gridCell', {
			anchorX: 0.5,
			anchorY: 0.5,
			x: gridStartX + col * cellSize,
			y: gridStartY + row * cellSize,
			alpha: 0.3
		});
		game.addChild(gridCell);
	}
}
// Initialize lawnmowers
for (var row = 0; row < gridRows; row++) {
	var lawnmower = new Lawnmower();
	lawnmower.x = 220;
	lawnmower.y = gridStartY + row * cellSize;
	game.addChild(lawnmower);
	lawnmowers.push(lawnmower);
}
function updateSunDisplay() {
	sunDisplay.setText('Sun: ' + sunPoints);
}
function canAffordPlant(plantType) {
	return sunPoints >= plantCosts[plantType];
}
function isSeedReady(plantType) {
	return seedCooldowns[plantType] <= 0;
}
function canPlant(plantType) {
	return canAffordPlant(plantType) && isSeedReady(plantType);
}
function getGridPosition(x, y) {
	var col = Math.floor((x - gridStartX + cellSize / 2) / cellSize);
	var row = Math.floor((y - gridStartY + cellSize / 2) / cellSize);
	if (col >= 0 && col < gridCols && row >= 0 && row < gridRows) {
		return {
			row: row,
			col: col,
			x: gridStartX + col * cellSize,
			y: gridStartY + row * cellSize
		};
	}
	return null;
}
function isGridOccupied(row, col) {
	for (var i = 0; i < plants.length; i++) {
		var plant = plants[i];
		var plantRow = Math.floor((plant.y - gridStartY + cellSize / 2) / cellSize);
		var plantCol = Math.floor((plant.x - gridStartX + cellSize / 2) / cellSize);
		if (plantRow === row && plantCol === col) {
			return true;
		}
	}
	return false;
}
function spawnZombie() {
	var rand = Math.random();
	var zombieType = 'normal';
	if (rand < 0.2) {
		zombieType = 'fast';
	} else if (rand < 0.35) {
		zombieType = 'doctor';
	} else if (rand < 0.45) {
		zombieType = 'snailBucket';
	}
	var zombie = new Zombie(zombieType);
	zombie.x = 2100;
	zombie.y = gridStartY + Math.floor(Math.random() * gridRows) * cellSize;
	game.addChild(zombie);
	zombies.push(zombie);
}
function gameOver() {
	if (gameState === 'playing') {
		gameState = 'lost';
		LK.showGameOver();
	}
}
function checkWinCondition() {
	if (currentWave > maxWaves && zombies.length === 0) {
		if (gameState === 'playing') {
			gameState = 'won';
			LK.showYouWin();
		}
	}
}
// Button event handlers
sunflowerButton.down = function (x, y, obj) {
	if (canPlant('sunflower')) {
		selectedPlantType = 'sunflower';
		shovelSelected = false;
		sunflowerButton.alpha = 1;
		sunflowerButton.tint = 0x808080;
		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
		peashooterButton.tint = 0xffffff;
		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
		wallnutButton.tint = 0xffffff;
		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
		potatomineButton.tint = 0xffffff;
		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
		carnivorousplantButton.tint = 0xffffff;
		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
		doomshroomButton.tint = 0xffffff;
		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
		repeaterButton.tint = 0xffffff;
		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
		threepeaterButton.tint = 0xffffff;
		shovelButton.alpha = 1;
		shovelButton.tint = 0xffffff;
	}
};
peashooterButton.down = function (x, y, obj) {
	if (canPlant('peashooter')) {
		selectedPlantType = 'peashooter';
		shovelSelected = false;
		peashooterButton.alpha = 1;
		peashooterButton.tint = 0x808080;
		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
		sunflowerButton.tint = 0xffffff;
		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
		wallnutButton.tint = 0xffffff;
		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
		potatomineButton.tint = 0xffffff;
		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
		carnivorousplantButton.tint = 0xffffff;
		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
		doomshroomButton.tint = 0xffffff;
		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
		repeaterButton.tint = 0xffffff;
		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
		threepeaterButton.tint = 0xffffff;
		shovelButton.alpha = 1;
		shovelButton.tint = 0xffffff;
	}
};
wallnutButton.down = function (x, y, obj) {
	if (canPlant('wallnut')) {
		selectedPlantType = 'wallnut';
		shovelSelected = false;
		wallnutButton.alpha = 1;
		wallnutButton.tint = 0x808080;
		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
		sunflowerButton.tint = 0xffffff;
		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
		peashooterButton.tint = 0xffffff;
		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
		potatomineButton.tint = 0xffffff;
		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
		carnivorousplantButton.tint = 0xffffff;
		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
		doomshroomButton.tint = 0xffffff;
		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
		repeaterButton.tint = 0xffffff;
		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
		threepeaterButton.tint = 0xffffff;
		shovelButton.alpha = 1;
		shovelButton.tint = 0xffffff;
	}
};
potatomineButton.down = function (x, y, obj) {
	if (canPlant('potatomine')) {
		selectedPlantType = 'potatomine';
		shovelSelected = false;
		potatomineButton.alpha = 1;
		potatomineButton.tint = 0x808080;
		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
		sunflowerButton.tint = 0xffffff;
		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
		peashooterButton.tint = 0xffffff;
		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
		wallnutButton.tint = 0xffffff;
		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
		carnivorousplantButton.tint = 0xffffff;
		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
		doomshroomButton.tint = 0xffffff;
		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
		repeaterButton.tint = 0xffffff;
		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
		threepeaterButton.tint = 0xffffff;
		shovelButton.alpha = 1;
		shovelButton.tint = 0xffffff;
	}
};
carnivorousplantButton.down = function (x, y, obj) {
	if (canPlant('carnivorousplant')) {
		selectedPlantType = 'carnivorousplant';
		shovelSelected = false;
		carnivorousplantButton.alpha = 1;
		carnivorousplantButton.tint = 0x808080;
		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
		sunflowerButton.tint = 0xffffff;
		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
		peashooterButton.tint = 0xffffff;
		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
		wallnutButton.tint = 0xffffff;
		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
		potatomineButton.tint = 0xffffff;
		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
		doomshroomButton.tint = 0xffffff;
		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
		repeaterButton.tint = 0xffffff;
		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
		threepeaterButton.tint = 0xffffff;
		shovelButton.alpha = 1;
		shovelButton.tint = 0xffffff;
	}
};
doomshroomButton.down = function (x, y, obj) {
	if (canPlant('doomshroom')) {
		selectedPlantType = 'doomshroom';
		shovelSelected = false;
		doomshroomButton.alpha = 1;
		doomshroomButton.tint = 0x808080;
		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
		sunflowerButton.tint = 0xffffff;
		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
		peashooterButton.tint = 0xffffff;
		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
		wallnutButton.tint = 0xffffff;
		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
		potatomineButton.tint = 0xffffff;
		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
		carnivorousplantButton.tint = 0xffffff;
		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
		repeaterButton.tint = 0xffffff;
		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
		threepeaterButton.tint = 0xffffff;
		shovelButton.alpha = 1;
		shovelButton.tint = 0xffffff;
	}
};
repeaterButton.down = function (x, y, obj) {
	if (canPlant('repeater')) {
		selectedPlantType = 'repeater';
		shovelSelected = false;
		repeaterButton.alpha = 1;
		repeaterButton.tint = 0x808080;
		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
		sunflowerButton.tint = 0xffffff;
		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
		peashooterButton.tint = 0xffffff;
		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
		wallnutButton.tint = 0xffffff;
		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
		potatomineButton.tint = 0xffffff;
		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
		carnivorousplantButton.tint = 0xffffff;
		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
		doomshroomButton.tint = 0xffffff;
		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
		threepeaterButton.tint = 0xffffff;
		shovelButton.alpha = 1;
		shovelButton.tint = 0xffffff;
	}
};
threepeaterButton.down = function (x, y, obj) {
	if (canPlant('threepeater')) {
		selectedPlantType = 'threepeater';
		shovelSelected = false;
		threepeaterButton.alpha = 1;
		threepeaterButton.tint = 0x808080;
		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
		sunflowerButton.tint = 0xffffff;
		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
		peashooterButton.tint = 0xffffff;
		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
		wallnutButton.tint = 0xffffff;
		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
		potatomineButton.tint = 0xffffff;
		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
		carnivorousplantButton.tint = 0xffffff;
		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
		doomshroomButton.tint = 0xffffff;
		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
		repeaterButton.tint = 0xffffff;
		shovelButton.alpha = 1;
		shovelButton.tint = 0xffffff;
	}
};
shovelButton.down = function (x, y, obj) {
	shovelSelected = true;
	selectedPlantType = null;
	shovelButton.alpha = 1;
	shovelButton.tint = 0x808080;
	sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
	sunflowerButton.tint = 0xffffff;
	peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
	peashooterButton.tint = 0xffffff;
	wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
	wallnutButton.tint = 0xffffff;
	potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
	potatomineButton.tint = 0xffffff;
	carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
	carnivorousplantButton.tint = 0xffffff;
	doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
	doomshroomButton.tint = 0xffffff;
	repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
	repeaterButton.tint = 0xffffff;
	threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
	threepeaterButton.tint = 0xffffff;
};
// Game event handlers
game.down = function (x, y, obj) {
	if (shovelSelected) {
		// Remove plant functionality
		var gridPos = getGridPosition(x, y);
		if (gridPos) {
			for (var i = plants.length - 1; i >= 0; i--) {
				var plant = plants[i];
				var plantRow = Math.floor((plant.y - gridStartY + cellSize / 2) / cellSize);
				var plantCol = Math.floor((plant.x - gridStartX + cellSize / 2) / cellSize);
				if (plantRow === gridPos.row && plantCol === gridPos.col) {
					plant.destroy();
					plants.splice(i, 1);
					shovelSelected = false;
					// Reset button alphas and tints
					sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
					sunflowerButton.tint = 0xffffff;
					peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
					peashooterButton.tint = 0xffffff;
					wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
					wallnutButton.tint = 0xffffff;
					potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
					potatomineButton.tint = 0xffffff;
					carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
					carnivorousplantButton.tint = 0xffffff;
					doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
					doomshroomButton.tint = 0xffffff;
					repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
					repeaterButton.tint = 0xffffff;
					threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
					threepeaterButton.tint = 0xffffff;
					shovelButton.alpha = 1;
					shovelButton.tint = 0xffffff;
					break;
				}
			}
		}
	} else if (selectedPlantType && canPlant(selectedPlantType)) {
		var gridPos = getGridPosition(x, y);
		if (gridPos && !isGridOccupied(gridPos.row, gridPos.col)) {
			sunPoints -= plantCosts[selectedPlantType];
			updateSunDisplay();
			// Start seed recharge
			seedCooldowns[selectedPlantType] = seedRechargeTime;
			var newPlant = new Plant(selectedPlantType);
			newPlant.x = gridPos.x;
			newPlant.y = gridPos.y;
			game.addChild(newPlant);
			plants.push(newPlant);
			LK.getSound('plant').play();
			selectedPlantType = null;
			// Reset button alphas and tints based on cooldown status
			sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
			sunflowerButton.tint = 0xffffff;
			peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
			peashooterButton.tint = 0xffffff;
			wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
			wallnutButton.tint = 0xffffff;
			potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
			potatomineButton.tint = 0xffffff;
			carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
			carnivorousplantButton.tint = 0xffffff;
			doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
			doomshroomButton.tint = 0xffffff;
			repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
			repeaterButton.tint = 0xffffff;
			threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
			threepeaterButton.tint = 0xffffff;
			shovelButton.alpha = 1;
			shovelButton.tint = 0xffffff;
		}
	}
};
// Main game update loop
game.update = function () {
	if (gameState !== 'playing') return;
	// Wave management
	waveSpawnTimer++;
	if (currentWave <= maxWaves) {
		if (waveSpawnTimer >= 1800 && zombiesInWave < maxZombiesInWave) {
			// Spawn every 30 seconds
			spawnZombie();
			zombiesInWave++;
			waveSpawnTimer = 0;
		} else if (zombiesInWave >= maxZombiesInWave && zombies.length === 0) {
			// Wave completed
			currentWave++;
			zombiesInWave = 0;
			maxZombiesInWave += 2; // Increase difficulty
			waveDisplay.setText('Wave: ' + currentWave + '/' + maxWaves);
		}
	}
	// Pea vs Zombie collision
	for (var i = peas.length - 1; i >= 0; i--) {
		var pea = peas[i];
		var peaHit = false;
		for (var j = zombies.length - 1; j >= 0; j--) {
			var zombie = zombies[j];
			if (Math.abs(pea.x - zombie.x) < 50 && Math.abs(pea.y - zombie.y) < 50) {
				if (zombie.takeDamage(pea.damage)) {
					zombie.destroy();
					zombies.splice(j, 1);
				}
				pea.destroy();
				peas.splice(i, 1);
				peaHit = true;
				break;
			}
		}
		if (!peaHit && pea.x > 2200) {
			pea.destroy();
			peas.splice(i, 1);
		}
	}
	// Remove destroyed suns
	for (var k = suns.length - 1; k >= 0; k--) {
		if (suns[k].collected) {
			suns.splice(k, 1);
		}
	}
	// Update seed cooldowns
	for (var seedType in seedCooldowns) {
		if (seedCooldowns[seedType] > 0) {
			seedCooldowns[seedType]--;
		}
	}
	// Lawnmower activation - check if zombies reach the left side
	for (var m = zombies.length - 1; m >= 0; m--) {
		var zombie = zombies[m];
		if (zombie.x < 250) {
			var zombieRow = Math.floor((zombie.y - gridStartY + cellSize / 2) / cellSize);
			if (zombieRow >= 0 && zombieRow < lawnmowers.length && !lawnmowers[zombieRow].activated) {
				lawnmowers[zombieRow].activated = true;
				// Remove zombie that triggered the lawnmower
				zombie.destroy();
				zombies.splice(m, 1);
			}
		}
	}
	// Remove lawnmowers that are off screen
	for (var n = lawnmowers.length - 1; n >= 0; n--) {
		if (lawnmowers[n].x > 2200) {
			lawnmowers.splice(n, 1);
		}
	}
	// Lawnmower vs zombie collision
	for (var p = 0; p < lawnmowers.length; p++) {
		var lawnmower = lawnmowers[p];
		if (lawnmower.activated) {
			for (var q = zombies.length - 1; q >= 0; q--) {
				var zombie = zombies[q];
				if (Math.abs(lawnmower.x - zombie.x) < 60 && Math.abs(lawnmower.y - zombie.y) < 60) {
					zombie.destroy();
					zombies.splice(q, 1);
				}
			}
		}
	}
	// Update button visual states based on cooldowns
	sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
	peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
	wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
	potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
	carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
	doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
	repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
	threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
	// Check win condition
	checkWinCondition();
};

===================================================================
--- original.js
+++ change.js
@@ -389,8 +389,25 @@
 		});
 		self.speed = -2;
 		self.health = 60;
 		self.maxHealth = 60;
+	} else if (self.type === 'doctor') {
+		zombieGraphics = self.attachAsset('zombieDoctor', {
+			anchorX: 0.5,
+			anchorY: 0.5
+		});
+		self.speed = -0.8;
+		self.health = 150;
+		self.maxHealth = 150;
+		self.healCooldown = 0;
+	} else if (self.type === 'snailBucket') {
+		zombieGraphics = self.attachAsset('snailBucket', {
+			anchorX: 0.5,
+			anchorY: 0.5
+		});
+		self.speed = -0.5;
+		self.health = 250;
+		self.maxHealth = 250;
 	} else {
 		zombieGraphics = self.attachAsset('zombie', {
 			anchorX: 0.5,
 			anchorY: 0.5
@@ -403,8 +420,40 @@
 		}
 		return false;
 	};
 	self.update = function () {
+		// Doctor zombie healing behavior
+		if (self.type === 'doctor') {
+			self.healCooldown++;
+			if (self.healCooldown >= 300) {
+				// Heal every 5 seconds
+				self.healCooldown = 0;
+				// Heal nearby zombies
+				for (var k = 0; k < zombies.length; k++) {
+					var nearbyZombie = zombies[k];
+					if (nearbyZombie !== self) {
+						var distance = Math.sqrt(Math.pow(self.x - nearbyZombie.x, 2) + Math.pow(self.y - nearbyZombie.y, 2));
+						if (distance < 150) {
+							nearbyZombie.health = Math.min(nearbyZombie.health + 30, nearbyZombie.maxHealth);
+							tween(nearbyZombie, {
+								tint: 0x00ff00
+							}, {
+								duration: 300,
+								easing: tween.easeOut,
+								onFinish: function onFinish() {
+									tween(nearbyZombie, {
+										tint: 0xffffff
+									}, {
+										duration: 300,
+										easing: tween.easeIn
+									});
+								}
+							});
+						}
+					}
+				}
+			}
+		}
 		if (!self.isAttacking) {
 			self.x += self.speed;
 			// Check if zombie reached the left side
 			if (self.x < 100) {
@@ -733,9 +782,18 @@
 	}
 	return false;
 }
 function spawnZombie() {
-	var zombie = new Zombie(Math.random() < 0.3 ? 'fast' : 'normal');
+	var rand = Math.random();
+	var zombieType = 'normal';
+	if (rand < 0.2) {
+		zombieType = 'fast';
+	} else if (rand < 0.35) {
+		zombieType = 'doctor';
+	} else if (rand < 0.45) {
+		zombieType = 'snailBucket';
+	}
+	var zombie = new Zombie(zombieType);
 	zombie.x = 2100;
 	zombie.y = gridStartY + Math.floor(Math.random() * gridRows) * cellSize;
 	game.addChild(zombie);
 	zombies.push(zombie);
@@ -759,135 +817,216 @@
 	if (canPlant('sunflower')) {
 		selectedPlantType = 'sunflower';
 		shovelSelected = false;
 		sunflowerButton.alpha = 1;
+		sunflowerButton.tint = 0x808080;
 		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+		peashooterButton.tint = 0xffffff;
 		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+		wallnutButton.tint = 0xffffff;
 		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+		potatomineButton.tint = 0xffffff;
 		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+		carnivorousplantButton.tint = 0xffffff;
 		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+		doomshroomButton.tint = 0xffffff;
 		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+		repeaterButton.tint = 0xffffff;
 		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+		threepeaterButton.tint = 0xffffff;
 		shovelButton.alpha = 1;
+		shovelButton.tint = 0xffffff;
 	}
 };
 peashooterButton.down = function (x, y, obj) {
 	if (canPlant('peashooter')) {
 		selectedPlantType = 'peashooter';
 		shovelSelected = false;
 		peashooterButton.alpha = 1;
+		peashooterButton.tint = 0x808080;
 		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+		sunflowerButton.tint = 0xffffff;
 		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+		wallnutButton.tint = 0xffffff;
 		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+		potatomineButton.tint = 0xffffff;
 		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+		carnivorousplantButton.tint = 0xffffff;
 		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+		doomshroomButton.tint = 0xffffff;
 		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+		repeaterButton.tint = 0xffffff;
 		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+		threepeaterButton.tint = 0xffffff;
 		shovelButton.alpha = 1;
+		shovelButton.tint = 0xffffff;
 	}
 };
 wallnutButton.down = function (x, y, obj) {
 	if (canPlant('wallnut')) {
 		selectedPlantType = 'wallnut';
 		shovelSelected = false;
 		wallnutButton.alpha = 1;
+		wallnutButton.tint = 0x808080;
 		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+		sunflowerButton.tint = 0xffffff;
 		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+		peashooterButton.tint = 0xffffff;
 		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+		potatomineButton.tint = 0xffffff;
 		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+		carnivorousplantButton.tint = 0xffffff;
 		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+		doomshroomButton.tint = 0xffffff;
 		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+		repeaterButton.tint = 0xffffff;
 		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+		threepeaterButton.tint = 0xffffff;
 		shovelButton.alpha = 1;
+		shovelButton.tint = 0xffffff;
 	}
 };
 potatomineButton.down = function (x, y, obj) {
 	if (canPlant('potatomine')) {
 		selectedPlantType = 'potatomine';
 		shovelSelected = false;
 		potatomineButton.alpha = 1;
+		potatomineButton.tint = 0x808080;
 		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+		sunflowerButton.tint = 0xffffff;
 		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+		peashooterButton.tint = 0xffffff;
 		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+		wallnutButton.tint = 0xffffff;
 		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+		carnivorousplantButton.tint = 0xffffff;
 		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+		doomshroomButton.tint = 0xffffff;
 		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+		repeaterButton.tint = 0xffffff;
 		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+		threepeaterButton.tint = 0xffffff;
 		shovelButton.alpha = 1;
+		shovelButton.tint = 0xffffff;
 	}
 };
 carnivorousplantButton.down = function (x, y, obj) {
 	if (canPlant('carnivorousplant')) {
 		selectedPlantType = 'carnivorousplant';
 		shovelSelected = false;
 		carnivorousplantButton.alpha = 1;
+		carnivorousplantButton.tint = 0x808080;
 		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+		sunflowerButton.tint = 0xffffff;
 		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+		peashooterButton.tint = 0xffffff;
 		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+		wallnutButton.tint = 0xffffff;
 		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+		potatomineButton.tint = 0xffffff;
 		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+		doomshroomButton.tint = 0xffffff;
 		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+		repeaterButton.tint = 0xffffff;
 		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+		threepeaterButton.tint = 0xffffff;
 		shovelButton.alpha = 1;
+		shovelButton.tint = 0xffffff;
 	}
 };
 doomshroomButton.down = function (x, y, obj) {
 	if (canPlant('doomshroom')) {
 		selectedPlantType = 'doomshroom';
 		shovelSelected = false;
 		doomshroomButton.alpha = 1;
+		doomshroomButton.tint = 0x808080;
 		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+		sunflowerButton.tint = 0xffffff;
 		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+		peashooterButton.tint = 0xffffff;
 		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+		wallnutButton.tint = 0xffffff;
 		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+		potatomineButton.tint = 0xffffff;
 		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+		carnivorousplantButton.tint = 0xffffff;
 		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+		repeaterButton.tint = 0xffffff;
 		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+		threepeaterButton.tint = 0xffffff;
 		shovelButton.alpha = 1;
+		shovelButton.tint = 0xffffff;
 	}
 };
 repeaterButton.down = function (x, y, obj) {
 	if (canPlant('repeater')) {
 		selectedPlantType = 'repeater';
 		shovelSelected = false;
 		repeaterButton.alpha = 1;
+		repeaterButton.tint = 0x808080;
 		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+		sunflowerButton.tint = 0xffffff;
 		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+		peashooterButton.tint = 0xffffff;
 		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+		wallnutButton.tint = 0xffffff;
 		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+		potatomineButton.tint = 0xffffff;
 		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+		carnivorousplantButton.tint = 0xffffff;
 		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+		doomshroomButton.tint = 0xffffff;
 		threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+		threepeaterButton.tint = 0xffffff;
 		shovelButton.alpha = 1;
+		shovelButton.tint = 0xffffff;
 	}
 };
 threepeaterButton.down = function (x, y, obj) {
 	if (canPlant('threepeater')) {
 		selectedPlantType = 'threepeater';
 		shovelSelected = false;
 		threepeaterButton.alpha = 1;
+		threepeaterButton.tint = 0x808080;
 		sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+		sunflowerButton.tint = 0xffffff;
 		peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+		peashooterButton.tint = 0xffffff;
 		wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+		wallnutButton.tint = 0xffffff;
 		potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+		potatomineButton.tint = 0xffffff;
 		carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+		carnivorousplantButton.tint = 0xffffff;
 		doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+		doomshroomButton.tint = 0xffffff;
 		repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+		repeaterButton.tint = 0xffffff;
 		shovelButton.alpha = 1;
+		shovelButton.tint = 0xffffff;
 	}
 };
 shovelButton.down = function (x, y, obj) {
 	shovelSelected = true;
 	selectedPlantType = null;
 	shovelButton.alpha = 1;
+	shovelButton.tint = 0x808080;
 	sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+	sunflowerButton.tint = 0xffffff;
 	peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+	peashooterButton.tint = 0xffffff;
 	wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+	wallnutButton.tint = 0xffffff;
 	potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+	potatomineButton.tint = 0xffffff;
 	carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+	carnivorousplantButton.tint = 0xffffff;
 	doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+	doomshroomButton.tint = 0xffffff;
 	repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+	repeaterButton.tint = 0xffffff;
 	threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+	threepeaterButton.tint = 0xffffff;
 };
 // Game event handlers
 game.down = function (x, y, obj) {
 	if (shovelSelected) {
@@ -901,18 +1040,27 @@
 				if (plantRow === gridPos.row && plantCol === gridPos.col) {
 					plant.destroy();
 					plants.splice(i, 1);
 					shovelSelected = false;
-					// Reset button alphas
+					// Reset button alphas and tints
 					sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+					sunflowerButton.tint = 0xffffff;
 					peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+					peashooterButton.tint = 0xffffff;
 					wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+					wallnutButton.tint = 0xffffff;
 					potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+					potatomineButton.tint = 0xffffff;
 					carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+					carnivorousplantButton.tint = 0xffffff;
 					doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+					doomshroomButton.tint = 0xffffff;
 					repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+					repeaterButton.tint = 0xffffff;
 					threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+					threepeaterButton.tint = 0xffffff;
 					shovelButton.alpha = 1;
+					shovelButton.tint = 0xffffff;
 					break;
 				}
 			}
 		}
@@ -929,18 +1077,27 @@
 			game.addChild(newPlant);
 			plants.push(newPlant);
 			LK.getSound('plant').play();
 			selectedPlantType = null;
-			// Reset button alphas based on cooldown status
+			// Reset button alphas and tints based on cooldown status
 			sunflowerButton.alpha = isSeedReady('sunflower') ? 1 : 0.5;
+			sunflowerButton.tint = 0xffffff;
 			peashooterButton.alpha = isSeedReady('peashooter') ? 1 : 0.5;
+			peashooterButton.tint = 0xffffff;
 			wallnutButton.alpha = isSeedReady('wallnut') ? 1 : 0.5;
+			wallnutButton.tint = 0xffffff;
 			potatomineButton.alpha = isSeedReady('potatomine') ? 1 : 0.5;
+			potatomineButton.tint = 0xffffff;
 			carnivorousplantButton.alpha = isSeedReady('carnivorousplant') ? 1 : 0.5;
+			carnivorousplantButton.tint = 0xffffff;
 			doomshroomButton.alpha = isSeedReady('doomshroom') ? 1 : 0.5;
+			doomshroomButton.tint = 0xffffff;
 			repeaterButton.alpha = isSeedReady('repeater') ? 1 : 0.5;
+			repeaterButton.tint = 0xffffff;
 			threepeaterButton.alpha = isSeedReady('threepeater') ? 1 : 0.5;
+			threepeaterButton.tint = 0xffffff;
 			shovelButton.alpha = 1;
+			shovelButton.tint = 0xffffff;
 		}
 	}
 };
 // Main game update loop