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

/**** 
* Classes
****/ 
var Bug = Container.expand(function (type, wave) {
	var self = Container.call(this);
	self.type = type || 'small';
	self.wave = wave || 1;
	self.isBoss = false;
	self.isFlying = false;
	self.isArmored = false;
	self.slowResistance = 0;
	self.pathIndex = 0;
	// Set base stats based on type
	switch (self.type) {
		case 'small':
			self.health = 15;
			self.speed = 2.5;
			self.reward = 5;
			break;
		case 'medium':
			self.health = 35;
			self.speed = 2;
			self.reward = 7;
			break;
		case 'large':
			self.health = 80;
			self.speed = 1.5;
			self.reward = 10;
			break;
		case 'fast':
			self.health = 20;
			self.speed = 4;
			self.reward = 8;
			break;
		case 'armored':
			self.health = 120;
			self.speed = 1;
			self.reward = 15;
			self.isArmored = true;
			break;
		case 'flying':
			self.health = 40;
			self.speed = 3;
			self.reward = 12;
			self.isFlying = true;
			break;
		case 'boss':
			self.health = 500;
			self.speed = 0.8;
			self.reward = 50;
			self.isBoss = true;
			self.slowResistance = 0.7;
			break;
	}
	// Scale with wave
	var waveMultiplier = 1 + (self.wave - 1) * 0.15;
	self.health = Math.floor(self.health * waveMultiplier);
	self.maxHealth = self.health;
	self.slowEffect = 1;
	self.poisonTimer = 0;
	var bugGraphics = self.attachAsset(self.type + 'Bug', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	// Boss scaling
	if (self.isBoss) {
		bugGraphics.scaleX = 1.5;
		bugGraphics.scaleY = 1.5;
		bugGraphics.tint = 0xFF4444;
	}
	// Flying effect
	if (self.isFlying) {
		bugGraphics.alpha = 0.8;
		var originalY = bugGraphics.y;
		tween(bugGraphics, {
			y: originalY - 10
		}, {
			duration: 1000,
			easing: tween.easeInOut,
			onFinish: function onFinish() {
				tween(bugGraphics, {
					y: originalY + 10
				}, {
					duration: 1000,
					easing: tween.easeInOut,
					onFinish: function onFinish() {
						// Create infinite loop
						tween(bugGraphics, {
							y: originalY - 10
						}, {
							duration: 1000,
							easing: tween.easeInOut,
							onFinish: arguments.callee.caller
						});
					}
				});
			}
		});
	}
	self.takeDamage = function (damage, damageType) {
		var actualDamage = damage;
		if (self.isArmored && damageType !== 'piercing') {
			actualDamage = Math.max(1, damage * 0.5);
		}
		if (damageType === 'poison') {
			self.poisonTimer = 180; // 3 seconds at 60fps
			actualDamage = damage * 0.3;
		}
		self.health -= actualDamage;
		var healthPercent = self.health / self.maxHealth;
		if (damageType === 'poison') {
			bugGraphics.tint = 0x00FF00;
		} else {
			bugGraphics.tint = healthPercent < 0.5 ? 0xFF6666 : 0xFFFFFF;
		}
		return self.health <= 0;
	};
	self.applySlow = function (slowAmount, duration) {
		if (self.slowResistance < 1) {
			var effectiveSlowAmount = slowAmount * (1 - self.slowResistance);
			self.slowEffect = Math.min(self.slowEffect, 1 - effectiveSlowAmount);
			LK.setTimeout(function () {
				self.slowEffect = 1;
			}, duration);
		}
	};
	self.update = function () {
		// Handle poison damage
		if (self.poisonTimer > 0) {
			self.poisonTimer--;
			if (self.poisonTimer % 30 === 0) {
				// Damage every 0.5 seconds
				self.takeDamage(5, 'poison');
			}
			if (self.poisonTimer <= 0) {
				bugGraphics.tint = 0xFFFFFF;
			}
		}
		// Check if bug is visible on screen and play appropriate sound
		var isOnScreen = self.x >= -50 && self.x <= 2098 && self.y >= -50 && self.y <= 2782;
		if (isOnScreen) {
			// Play bug sound based on type every 2 seconds (120 frames)
			if (LK.ticks % 120 === 0) {
				if (self.isFlying) {
					var flyingSound = LK.getSound('bugflying');
					if (flyingSound) flyingSound.play();
				} else {
					var crawlingSound = LK.getSound('bugcrawling');
					if (crawlingSound) crawlingSound.play();
				}
			}
		}
		// Move along path
		if (self.pathIndex < gamePath.length) {
			var target = gamePath[self.pathIndex];
			var dx = target.x - self.x;
			var dy = target.y - self.y;
			var distance = Math.sqrt(dx * dx + dy * dy);
			if (distance < 30) {
				self.pathIndex++;
			} else {
				var actualSpeed = self.speed * self.slowEffect;
				self.x += dx / distance * actualSpeed;
				self.y += dy / distance * actualSpeed;
			}
		}
	};
	return self;
});
var Bullet = Container.expand(function (bulletType, tower) {
	var self = Container.call(this);
	self.speed = 8;
	self.target = null;
	self.damage = 1;
	self.type = bulletType || 'basic';
	self.tower = tower;
	self.splashRadius = 0;
	self.piercing = false;
	var bulletGraphics = self.attachAsset('bullet', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	// Customize bullet based on type
	switch (self.type) {
		case 'rapid':
			bulletGraphics.tint = 0x00AAFF;
			bulletGraphics.scaleX = 0.7;
			bulletGraphics.scaleY = 0.7;
			self.speed = 12;
			break;
		case 'sniper':
			bulletGraphics.tint = 0xFF4444;
			bulletGraphics.scaleX = 1.2;
			bulletGraphics.scaleY = 1.2;
			self.speed = 15;
			self.piercing = true;
			break;
		case 'cannon':
			bulletGraphics.tint = 0xFF8800;
			bulletGraphics.scaleX = 1.5;
			bulletGraphics.scaleY = 1.5;
			self.speed = 6;
			self.splashRadius = 100;
			break;
		case 'freeze':
			bulletGraphics.tint = 0x88CCFF;
			self.speed = 10;
			break;
		case 'poison':
			bulletGraphics.tint = 0x00FF00;
			self.speed = 9;
			break;
	}
	self.update = function () {
		if (!self.target || !self.target.parent) {
			self.destroy();
			return;
		}
		var dx = self.target.x - self.x;
		var dy = self.target.y - self.y;
		var distance = Math.sqrt(dx * dx + dy * dy);
		if (distance < 20) {
			self.hitTarget();
			self.destroy();
		} else {
			self.x += dx / distance * self.speed;
			self.y += dy / distance * self.speed;
		}
	};
	self.hitTarget = function () {
		var damageType = self.type === 'sniper' ? 'piercing' : self.type === 'poison' ? 'poison' : 'normal';
		// Apply main damage
		if (self.target.takeDamage(self.damage, damageType)) {
			// Play bugdeath sound immediately when bug dies
			var bugDeathSound = LK.getSound('bugDeath');
			if (bugDeathSound) bugDeathSound.play();
			self.killBug(self.target);
		}
		// Special effects
		if (self.type === 'freeze') {
			self.target.applySlow(0.6, 2000); // 60% slow for 2 seconds
		}
		// Splash damage
		if (self.splashRadius > 0) {
			for (var i = 0; i < bugs.length; i++) {
				var bug = bugs[i];
				if (bug !== self.target) {
					var dx = bug.x - self.target.x;
					var dy = bug.y - self.target.y;
					var dist = Math.sqrt(dx * dx + dy * dy);
					if (dist <= self.splashRadius) {
						if (bug.takeDamage(self.damage * 0.6, damageType)) {
							// Play bugdeath sound for splash damage kills too
							var bugDeathSound = LK.getSound('bugDeath');
							if (bugDeathSound) bugDeathSound.play();
							self.killBug(bug);
						}
					}
				}
			}
		}
	};
	self.killBug = function (bug) {
		coins += bug.reward;
		bugsKilled++;
		updateUI();
		for (var i = bugs.length - 1; i >= 0; i--) {
			if (bugs[i] === bug) {
				bug.destroy();
				bugs.splice(i, 1);
				break;
			}
		}
	};
	return self;
});
var Colonist = Container.expand(function () {
	var self = Container.call(this);
	self.speed = 1;
	self.pathIndex = 0;
	self.isSafe = false;
	var colonistGraphics = self.attachAsset('colonist', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.update = function () {
		if (self.isSafe) return;
		if (self.pathIndex < colonistPath.length) {
			var target = colonistPath[self.pathIndex];
			var dx = target.x - self.x;
			var dy = target.y - self.y;
			var distance = Math.sqrt(dx * dx + dy * dy);
			if (distance < 20) {
				self.pathIndex++;
				if (self.pathIndex >= colonistPath.length) {
					self.isSafe = true;
					colonistsSaved++;
					updateUI();
				}
			} else {
				self.x += dx / distance * self.speed;
				self.y += dy / distance * self.speed;
			}
		}
	};
	return self;
});
var MainMenu = Container.expand(function () {
	var self = Container.call(this);
	// Main menu background
	var menuBg = self.attachAsset('background', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	menuBg.x = 1024;
	menuBg.y = 1366;
	menuBg.width = 2048;
	menuBg.height = 2732;
	menuBg.alpha = 0.7;
	// Title text
	var titleText = new Text2('COLONY DEFENSE', {
		size: 120,
		fill: 0xFFFFFF
	});
	titleText.anchor.set(0.5, 0.5);
	titleText.x = 1024;
	titleText.y = 800;
	self.addChild(titleText);
	// Subtitle text
	var subtitleText = new Text2('Defend the colonies from alien bugs!', {
		size: 60,
		fill: 0xAADDFF
	});
	subtitleText.anchor.set(0.5, 0.5);
	subtitleText.x = 1024;
	subtitleText.y = 950;
	self.addChild(subtitleText);
	// Start button
	var startButton = self.attachAsset('bunker', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	startButton.x = 1024;
	startButton.y = 1400;
	startButton.width = 300;
	startButton.height = 200;
	startButton.tint = 0x00AA00;
	var startText = new Text2('START GAME', {
		size: 50,
		fill: 0xFFFFFF
	});
	startText.anchor.set(0.5, 0.5);
	startText.x = 1024;
	startText.y = 1400;
	self.addChild(startText);
	// Instructions text
	var instructText = new Text2('Use robot towers to stop the alien invasion!\nSave the colonists and defend your colonies!', {
		size: 40,
		fill: 0xCCCCCC
	});
	instructText.anchor.set(0.5, 0.5);
	instructText.x = 1024;
	instructText.y = 1800;
	self.addChild(instructText);
	// Add pulsing animation to start button
	tween(startButton, {
		scaleX: 1.1,
		scaleY: 1.1
	}, {
		duration: 1000,
		easing: tween.easeInOut,
		onFinish: function onFinish() {
			tween(startButton, {
				scaleX: 1,
				scaleY: 1
			}, {
				duration: 1000,
				easing: tween.easeInOut,
				onFinish: arguments.callee.caller
			});
		}
	});
	self.down = function (x, y, obj) {
		// Check if clicked on start button area
		var dx = x - startButton.x;
		var dy = y - startButton.y;
		var distance = Math.sqrt(dx * dx + dy * dy);
		if (distance <= 150) {
			startGame();
		}
	};
	return self;
});
var Tower = Container.expand(function (towerType) {
	var self = Container.call(this);
	self.type = towerType || 'basic';
	self.range = 150;
	self.damage = 1;
	self.fireRate = 30;
	self.lastShot = 0;
	self.level = 1;
	self.maxLevel = 5;
	self.upgradeCost = 20;
	self.sellValue = 15;
	// Set base stats based on tower type
	switch (self.type) {
		case 'basic':
			self.range = 150;
			self.damage = 8;
			self.fireRate = 40;
			self.upgradeCost = 20;
			break;
		case 'rapid':
			self.range = 120;
			self.damage = 4;
			self.fireRate = 15;
			self.upgradeCost = 30;
			break;
		case 'sniper':
			self.range = 250;
			self.damage = 25;
			self.fireRate = 80;
			self.upgradeCost = 50;
			break;
		case 'cannon':
			self.range = 140;
			self.damage = 20;
			self.fireRate = 60;
			self.upgradeCost = 40;
			break;
		case 'freeze':
			self.range = 130;
			self.damage = 6;
			self.fireRate = 35;
			self.upgradeCost = 35;
			break;
		case 'poison':
			self.range = 140;
			self.damage = 12;
			self.fireRate = 45;
			self.upgradeCost = 45;
			break;
	}
	var towerGraphics = self.attachAsset('tower', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	// Set tower color based on type
	switch (self.type) {
		case 'rapid':
			towerGraphics.tint = 0x00AAFF;
			break;
		case 'sniper':
			towerGraphics.tint = 0xFF4444;
			break;
		case 'cannon':
			towerGraphics.tint = 0xFF8800;
			break;
		case 'freeze':
			towerGraphics.tint = 0x88CCFF;
			break;
		case 'poison':
			towerGraphics.tint = 0x00FF00;
			break;
		default:
			towerGraphics.tint = 0x4169E1;
	}
	self.canShoot = function () {
		return LK.ticks - self.lastShot >= self.fireRate;
	};
	self.findTarget = function () {
		var bestTarget = null;
		var bestScore = -1;
		for (var i = 0; i < bugs.length; i++) {
			var bug = bugs[i];
			var dx = bug.x - self.x;
			var dy = bug.y - self.y;
			var distance = Math.sqrt(dx * dx + dy * dy);
			if (distance <= self.range) {
				// Prioritize based on progress along path and bug type
				var score = bug.pathIndex * 1000 + (bug.isBoss ? 500 : 0) - distance;
				if (score > bestScore) {
					bestScore = score;
					bestTarget = bug;
				}
			}
		}
		return bestTarget;
	};
	self.shoot = function (target) {
		if (!self.canShoot()) return;
		var bullet = new Bullet(self.type, self);
		bullet.x = self.x;
		bullet.y = self.y;
		bullet.target = target;
		bullet.damage = self.damage;
		bullets.push(bullet);
		game.addChild(bullet);
		self.lastShot = LK.ticks;
		var shootSound = LK.getSound('shoot');
		if (shootSound) shootSound.play();
		// Visual muzzle flash
		towerGraphics.alpha = 1.5;
		tween(towerGraphics, {
			alpha: 1
		}, {
			duration: 100
		});
	};
	self.upgrade = function () {
		if (self.level >= self.maxLevel) return false;
		if (coins >= self.upgradeCost) {
			coins -= self.upgradeCost;
			self.level++;
			// Upgrade stats
			self.damage = Math.floor(self.damage * 1.3);
			self.range += 15;
			if (self.type === 'rapid') {
				self.fireRate = Math.max(8, self.fireRate - 3);
			} else {
				self.fireRate = Math.max(15, self.fireRate - 5);
			}
			self.sellValue += Math.floor(self.upgradeCost * 0.7);
			self.upgradeCost = Math.floor(self.upgradeCost * 1.5);
			towerGraphics.tint = 0x00FF00;
			tween(towerGraphics, {
				tint: self.type === 'rapid' ? 0x00AAFF : self.type === 'sniper' ? 0xFF4444 : self.type === 'cannon' ? 0xFF8800 : self.type === 'freeze' ? 0x88CCFF : self.type === 'poison' ? 0x00FF00 : 0x4169E1
			}, {
				duration: 500
			});
			updateUI();
			return true;
		}
		return false;
	};
	self.sell = function () {
		coins += self.sellValue;
		selectedTower = null;
		for (var i = towers.length - 1; i >= 0; i--) {
			if (towers[i] === self) {
				towers.splice(i, 1);
				break;
			}
		}
		self.destroy();
		updateUI();
	};
	self.update = function () {
		var target = self.findTarget();
		if (target) {
			self.shoot(target);
		}
	};
	self.down = function (x, y, obj) {
		selectedTower = self;
		showTowerInfo();
	};
	return self;
});
var TowerButton = Container.expand(function (towerType, cost) {
	var self = Container.call(this);
	self.towerType = towerType;
	self.cost = cost;
	var buttonBg = self.attachAsset('tower', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	buttonBg.width = 120;
	buttonBg.height = 120;
	// Set button color based on tower type
	switch (towerType) {
		case 'rapid':
			buttonBg.tint = 0x00AAFF;
			break;
		case 'sniper':
			buttonBg.tint = 0xFF4444;
			break;
		case 'cannon':
			buttonBg.tint = 0xFF8800;
			break;
		case 'freeze':
			buttonBg.tint = 0x88CCFF;
			break;
		case 'poison':
			buttonBg.tint = 0x00FF00;
			break;
		default:
			buttonBg.tint = 0x4169E1;
	}
	var costText = new Text2('$' + cost, {
		size: 30,
		fill: 0xFFFFFF
	});
	costText.anchor.set(0.5, 0.5);
	costText.y = 50;
	self.addChild(costText);
	var typeText = new Text2(towerType.toUpperCase(), {
		size: 25,
		fill: 0xFFFFFF
	});
	typeText.anchor.set(0.5, 0.5);
	typeText.y = -50;
	self.addChild(typeText);
	self.update = function () {
		var canAfford = coins >= self.cost;
		self.alpha = canAfford ? 1 : 0.5;
	};
	self.down = function (x, y, obj) {
		if (coins >= self.cost) {
			placingTowerType = self.towerType;
			selectedTower = null;
			updateUI();
		}
	};
	return self;
});

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

/**** 
* Game Code
****/ 
var bugs = [];
var towers = [];
var bullets = [];
var colonists = [];
var coins = 80;
var wave = 1;
var maxWaves = 250;
var bugsKilled = 0;
var colonistsSaved = 0;
var selectedTower = null;
var placingTowerType = null;
var waveTimer = 0;
var waveDelay = 300;
var bugsToSpawn = 5;
var bugsSpawned = 0;
var gameWon = false;
var towerButtons = [];
var gameState = 'menu'; // 'menu' or 'playing'
var mainMenu = null;
var gameElements = [];
// Tower costs
var towerCosts = {
	basic: 15,
	rapid: 25,
	sniper: 40,
	cannon: 35,
	freeze: 30,
	poison: 45
};
// Game path for bugs
var gamePath = [{
	x: 0,
	y: 1366
}, {
	x: 400,
	y: 1366
}, {
	x: 400,
	y: 800
}, {
	x: 800,
	y: 800
}, {
	x: 800,
	y: 1366
}, {
	x: 1200,
	y: 1366
}, {
	x: 1200,
	y: 600
}, {
	x: 1600,
	y: 600
}, {
	x: 1600,
	y: 1000
}, {
	x: 2048,
	y: 1000
}];
// Colonist path to bunker - updated to connect colonies better
var colonistPath = [{
	x: 500,
	// Start near colony1
	y: 700
}, {
	x: 900,
	y: 900
}, {
	x: 1200,
	y: 900
}, {
	x: 1400,
	y: 800
}, {
	x: 1600,
	y: 1000
}, {
	x: 1800,
	y: 1400
}];
// Create colonies array and bunker variable
var colonies = [];
var bunker = null;
// Initialize main menu
mainMenu = new MainMenu();
game.addChild(mainMenu);
// UI Elements
var coinText = new Text2('Coins: 50', {
	size: 60,
	fill: 0xFFD700
});
coinText.anchor.set(0, 0);
LK.gui.topRight.addChild(coinText);
var waveText = new Text2('Wave: 1', {
	size: 60,
	fill: 0xFFFFFF
});
waveText.anchor.set(0.5, 0);
LK.gui.top.addChild(waveText);
var infoText = new Text2('Tap empty space to place tower ($15)', {
	size: 40,
	fill: 0xFFFFFF
});
infoText.anchor.set(0.5, 1);
LK.gui.bottom.addChild(infoText);
var killsText = new Text2('Bugs Killed: 0', {
	size: 50,
	fill: 0xFF6666
});
killsText.anchor.set(0, 0);
LK.gui.left.addChild(killsText);
var savedText = new Text2('Colonists Saved: 0', {
	size: 50,
	fill: 0x00FF00
});
savedText.anchor.set(0, 0);
savedText.y = 80;
LK.gui.left.addChild(savedText);
// Create tower selection buttons
var towerTypes = ['basic', 'rapid', 'sniper', 'cannon', 'freeze', 'poison'];
var buttonSpacing = 140;
var startX = 100;
for (var i = 0; i < towerTypes.length; i++) {
	var towerButton = new TowerButton(towerTypes[i], towerCosts[towerTypes[i]]);
	towerButton.x = startX + i * buttonSpacing;
	towerButton.y = 2600;
	game.addChild(towerButton);
	towerButtons.push(towerButton);
}
function startGame() {
	if (gameState === 'playing') return;
	gameState = 'playing';
	// Hide main menu
	if (mainMenu) {
		mainMenu.destroy();
		mainMenu = null;
	}
	// Reset game variables
	bugs = [];
	towers = [];
	bullets = [];
	colonists = [];
	colonies = [];
	coins = 80;
	wave = 1;
	bugsKilled = 0;
	colonistsSaved = 0;
	selectedTower = null;
	placingTowerType = null;
	waveTimer = 0;
	bugsToSpawn = 5;
	bugsSpawned = 0;
	gameWon = false;
	towerButtons = [];
	// Initialize game elements
	initializeGameElements();
	// Start first wave after delay
	LK.setTimeout(function () {
		startNewWave();
	}, 3000);
}
function initializeGameElements() {
	// Add background image
	var backgroundImage = game.addChild(LK.getAsset('background', {
		anchorX: 0.5,
		anchorY: 0.5
	}));
	backgroundImage.x = 1024;
	backgroundImage.y = 1366;
	backgroundImage.width = 2048;
	backgroundImage.height = 2732;
	gameElements.push(backgroundImage);
	// Create initial colony
	var colony1 = game.addChild(LK.getAsset('colony', {
		anchorX: 0.5,
		anchorY: 0.5
	}));
	colony1.x = 400;
	colony1.y = 600;
	colonies.push(colony1);
	gameElements.push(colony1);
	// Create bunker
	bunker = game.addChild(LK.getAsset('bunker', {
		anchorX: 0.5,
		anchorY: 0.5
	}));
	bunker.x = 1800;
	bunker.y = 1400;
	gameElements.push(bunker);
	gameElements.push(towerButton);
	// Show initial instructions
	var instructionText = new Text2('Defend the colonies from alien bugs!\nUse robot towers to stop them!', {
		size: 60,
		fill: 0xFFFFFF
	});
	instructionText.anchor.set(0.5, 0.5);
	instructionText.x = 1024;
	instructionText.y = 600;
	game.addChild(instructionText);
	gameElements.push(instructionText);
	tween(instructionText, {
		alpha: 0
	}, {
		duration: 4000,
		onFinish: function onFinish() {
			instructionText.destroy();
		}
	});
	// Play background music
	var bgMusic = LK.getMusic ? LK.getMusic('gameMusic') : null;
	if (bgMusic || LK.playMusic) {
		LK.playMusic('gameMusic');
	}
}
function updateUI() {
	if (gameState !== 'playing') return;
	coinText.setText('Coins: ' + coins);
	killsText.setText('Bugs Killed: ' + bugsKilled);
	savedText.setText('Colonists Saved: ' + colonistsSaved);
	if (selectedTower) {
		var upgradeText = selectedTower.level >= selectedTower.maxLevel ? 'MAX LEVEL' : 'Upgrade ($' + selectedTower.upgradeCost + ')';
		infoText.setText('Tower Lv.' + selectedTower.level + ' - ' + upgradeText + ' | Sell ($' + selectedTower.sellValue + ')');
	} else if (placingTowerType) {
		infoText.setText('Placing ' + placingTowerType.toUpperCase() + ' tower ($' + towerCosts[placingTowerType] + ') - Tap location');
	} else {
		infoText.setText('Select a tower type below to place');
	}
}
function showTowerInfo() {
	updateUI();
}
function spawnBug() {
	if (bugsSpawned >= bugsToSpawn) return;
	var bugType;
	var isBossWave = wave % 5 === 0;
	if (isBossWave && bugsSpawned === 0) {
		bugType = 'boss';
	} else if (wave >= 15) {
		var advancedTypes = ['medium', 'large', 'fast', 'armored', 'flying'];
		bugType = advancedTypes[Math.floor(Math.random() * advancedTypes.length)];
	} else if (wave >= 10) {
		var midTypes = ['small', 'medium', 'fast', 'flying'];
		bugType = midTypes[Math.floor(Math.random() * midTypes.length)];
	} else if (wave >= 5) {
		var earlyTypes = ['small', 'medium', 'fast'];
		bugType = earlyTypes[Math.floor(Math.random() * earlyTypes.length)];
	} else {
		var basicTypes = ['small', 'small', 'medium'];
		bugType = basicTypes[Math.floor(Math.random() * basicTypes.length)];
	}
	var bug = new Bug(bugType, wave);
	bug.x = gamePath[0].x;
	bug.y = gamePath[0].y;
	bugs.push(bug);
	game.addChild(bug);
	bugsSpawned++;
	// Play bug crawling sound when bug spawns
	var crawlingSound = LK.getSound('bugcrawling');
	if (crawlingSound) crawlingSound.play();
}
function spawnNewColony() {
	// Only spawn if we have fewer than 8 colonies total
	if (colonies.length >= 8) return;
	// Generate random position for new colony in safe areas
	var attempts = 0;
	var newColony = null;
	while (attempts < 30 && !newColony) {
		var x = 200 + Math.random() * 1600; // Random X between 200-1800
		var y = 400 + Math.random() * 1000; // Random Y between 400-1400
		var canPlace = true;
		attempts++;
		// Ensure we're not too close to screen edges
		if (x < 150 || x > 1898 || y < 150 || y > 2582) {
			canPlace = false;
			continue;
		}
		// Check distance from path
		for (var i = 0; i < gamePath.length; i++) {
			var dx = x - gamePath[i].x;
			var dy = y - gamePath[i].y;
			if (Math.sqrt(dx * dx + dy * dy) < 180) {
				canPlace = false;
				break;
			}
		}
		if (!canPlace) continue;
		// Check distance from existing colonies and bunker
		var structures = colonies.slice();
		structures.push(bunker);
		for (var i = 0; i < structures.length; i++) {
			var dx = x - structures[i].x;
			var dy = y - structures[i].y;
			if (Math.sqrt(dx * dx + dy * dy) < 250) {
				canPlace = false;
				break;
			}
		}
		if (!canPlace) continue;
		// Check distance from existing towers
		for (var i = 0; i < towers.length; i++) {
			var dx = x - towers[i].x;
			var dy = y - towers[i].y;
			if (Math.sqrt(dx * dx + dy * dy) < 180) {
				canPlace = false;
				break;
			}
		}
		if (canPlace) {
			newColony = game.addChild(LK.getAsset('colony', {
				anchorX: 0.5,
				anchorY: 0.5
			}));
			newColony.x = x;
			newColony.y = y;
			colonies.push(newColony);
			// Flash effect for new colony
			LK.effects.flashObject(newColony, 0x00FF00, 1000);
			break;
		}
	}
}
function startNewWave() {
	// Only increment wave if this isn't the initial call
	if (wave === 1 && bugsSpawned > 0) {
		wave++;
	} else if (wave > 1) {
		wave++;
	}
	// Add new colony each wave (starting from wave 2 since we start with only 1 colony)
	if (wave >= 2) {
		spawnNewColony();
	}
	var isBossWave = wave % 5 === 0;
	bugsToSpawn = isBossWave ? 1 : Math.min(15, 3 + wave * 2);
	bugsSpawned = 0;
	waveTimer = 0;
	waveText.setText('Wave: ' + wave + (isBossWave ? ' (BOSS)' : '') + ' | Colonies: ' + colonies.length);
	if (isBossWave) {
		// Show boss warning
		var bossWarning = new Text2('BOSS WAVE INCOMING!', {
			size: 80,
			fill: 0xFF4444
		});
		bossWarning.anchor.set(0.5, 0.5);
		bossWarning.x = 1024;
		bossWarning.y = 800;
		game.addChild(bossWarning);
		tween(bossWarning, {
			alpha: 0
		}, {
			duration: 3000,
			onFinish: function onFinish() {
				bossWarning.destroy();
			}
		});
	}
	// Spawn fleeing colonists and male colonists with dramatic animations
	if (wave <= 15 && Math.random() < 0.7) {
		// Spawn regular colonists fleeing from random colonies
		var colonistsToSpawn = Math.min(colonies.length, Math.ceil(wave / 2));
		for (var i = 0; i < colonistsToSpawn; i++) {
			var randomColony = colonies[Math.floor(Math.random() * colonies.length)];
			var colonist = new Colonist();
			colonist.x = randomColony.x + (Math.random() - 0.5) * 150;
			colonist.y = randomColony.y + (Math.random() - 0.5) * 150;
			colonists.push(colonist);
			game.addChild(colonist);
			// Add fleeing animation - quick movement toward first waypoint
			tween(colonist, {
				x: colonistPath[0].x + (Math.random() - 0.5) * 50,
				y: colonistPath[0].y + (Math.random() - 0.5) * 50
			}, {
				duration: 1000 + Math.random() * 1000,
				easing: tween.easeOut
			});
		}
		// Spawn male colonists fleeing from random colonies
		if (Math.random() < 0.5 && colonies.length > 1) {
			var randomColony = colonies[Math.floor(Math.random() * colonies.length)];
			var maleColonist = game.addChild(LK.getAsset('colonistmale', {
				anchorX: 0.5,
				anchorY: 0.5
			}));
			maleColonist.x = randomColony.x + (Math.random() - 0.5) * 150;
			maleColonist.y = randomColony.y + (Math.random() - 0.5) * 150;
			maleColonist.speed = 1.2;
			maleColonist.pathIndex = 0;
			maleColonist.isSafe = false;
			maleColonist.isMale = true;
			// Add to colonists array for tracking
			colonists.push(maleColonist);
			// Add panic animation - shake then flee
			tween(maleColonist, {
				x: maleColonist.x + 20
			}, {
				duration: 100,
				easing: tween.easeInOut,
				onFinish: function onFinish() {
					tween(maleColonist, {
						x: maleColonist.x - 40
					}, {
						duration: 100,
						easing: tween.easeInOut,
						onFinish: function onFinish() {
							tween(maleColonist, {
								x: colonistPath[Math.floor(colonistPath.length / 2)].x,
								y: colonistPath[Math.floor(colonistPath.length / 2)].y
							}, {
								duration: 2000 + Math.random() * 1500,
								easing: tween.easeOut
							});
						}
					});
				}
			});
			// Add custom update method for male colonist
			maleColonist.update = function () {
				if (maleColonist.isSafe) return;
				if (maleColonist.pathIndex < colonistPath.length) {
					var target = colonistPath[maleColonist.pathIndex];
					var dx = target.x - maleColonist.x;
					var dy = target.y - maleColonist.y;
					var distance = Math.sqrt(dx * dx + dy * dy);
					if (distance < 20) {
						maleColonist.pathIndex++;
						if (maleColonist.pathIndex >= colonistPath.length) {
							maleColonist.isSafe = true;
							colonistsSaved++;
							updateUI();
						}
					} else {
						maleColonist.x += dx / distance * maleColonist.speed;
						maleColonist.y += dy / distance * maleColonist.speed;
					}
				}
			};
		}
	}
}
game.down = function (x, y, obj) {
	if (gameState !== 'playing') return;
	// Handle tower upgrade/sell if tower is selected
	if (selectedTower && x > 1500 && y > 2400) {
		// Clicked in UI area
		if (y > 2500) {
			// Sell button area
			selectedTower.sell();
		} else {
			// Upgrade button area
			selectedTower.upgrade();
		}
		updateUI();
		return;
	}
	// Handle tower placement
	if (placingTowerType && y < 2400) {
		var canPlace = true;
		var cost = towerCosts[placingTowerType];
		// Check if we have enough coins
		if (coins < cost) {
			canPlace = false;
		}
		// Check distance from path
		for (var i = 0; i < gamePath.length; i++) {
			var dx = x - gamePath[i].x;
			var dy = y - gamePath[i].y;
			if (Math.sqrt(dx * dx + dy * dy) < 80) {
				canPlace = false;
				break;
			}
		}
		// Check distance from existing towers
		for (var i = 0; i < towers.length; i++) {
			var dx = x - towers[i].x;
			var dy = y - towers[i].y;
			if (Math.sqrt(dx * dx + dy * dy) < 100) {
				canPlace = false;
				break;
			}
		}
		// Check distance from colonies and bunker
		var structures = colonies.slice(); // Copy colonies array
		structures.push(bunker);
		for (var i = 0; i < structures.length; i++) {
			var dx = x - structures[i].x;
			var dy = y - structures[i].y;
			if (Math.sqrt(dx * dx + dy * dy) < 120) {
				canPlace = false;
				break;
			}
		}
		if (canPlace) {
			var tower = new Tower(placingTowerType);
			tower.x = x;
			tower.y = y;
			towers.push(tower);
			game.addChild(tower);
			coins -= cost;
			placingTowerType = null;
			updateUI();
		}
	} else {
		// Clear selection
		selectedTower = null;
		placingTowerType = null;
	}
	updateUI();
};
game.update = function () {
	if (gameState !== 'playing') return;
	if (gameWon) return;
	// Spawn bugs
	waveTimer++;
	if (waveTimer % 45 === 0 && bugsSpawned < bugsToSpawn) {
		spawnBug();
	}
	// Check if wave is complete
	if (bugsSpawned >= bugsToSpawn && bugs.length === 0) {
		if (wave >= maxWaves) {
			gameWon = true;
			LK.setTimeout(function () {
				LK.showYouWin();
			}, 1000);
		} else {
			LK.setTimeout(function () {
				startNewWave();
			}, 2000);
		}
	}
	// Clean up bullets
	for (var i = bullets.length - 1; i >= 0; i--) {
		if (!bullets[i].parent) {
			bullets.splice(i, 1);
		}
	}
	// Check for bug-colonist intersections
	for (var i = bugs.length - 1; i >= 0; i--) {
		var bug = bugs[i];
		for (var j = colonists.length - 1; j >= 0; j--) {
			var colonist = colonists[j];
			if (!colonist.isSafe && colonist.intersects && colonist.intersects(bug)) {
				// Play colonist scream sound when killed by bug
				var screamSound = LK.getSound('colonistscream');
				if (screamSound) screamSound.play();
				// Remove the colonist
				colonist.destroy();
				colonists.splice(j, 1);
				break; // One bug can only kill one colonist per frame
			}
		}
	}
	// Check for bugs reaching the end
	var bugsReachedEnd = 0;
	for (var i = bugs.length - 1; i >= 0; i--) {
		var bug = bugs[i];
		if (bug.pathIndex >= gamePath.length) {
			bug.destroy();
			bugs.splice(i, 1);
			bugsReachedEnd++;
			if (bug.isBoss) {
				bugsReachedEnd += 4; // Boss counts as 5 bugs reaching end
			}
		}
	}
	// Check lose condition
	if (bugsReachedEnd >= 3) {
		LK.showGameOver();
	}
	// Update UI periodically
	if (LK.ticks % 60 === 0) {
		updateUI();
	}
};

===================================================================
--- original.js
+++ change.js
@@ -310,8 +310,94 @@
 		}
 	};
 	return self;
 });
+var MainMenu = Container.expand(function () {
+	var self = Container.call(this);
+	// Main menu background
+	var menuBg = self.attachAsset('background', {
+		anchorX: 0.5,
+		anchorY: 0.5
+	});
+	menuBg.x = 1024;
+	menuBg.y = 1366;
+	menuBg.width = 2048;
+	menuBg.height = 2732;
+	menuBg.alpha = 0.7;
+	// Title text
+	var titleText = new Text2('COLONY DEFENSE', {
+		size: 120,
+		fill: 0xFFFFFF
+	});
+	titleText.anchor.set(0.5, 0.5);
+	titleText.x = 1024;
+	titleText.y = 800;
+	self.addChild(titleText);
+	// Subtitle text
+	var subtitleText = new Text2('Defend the colonies from alien bugs!', {
+		size: 60,
+		fill: 0xAADDFF
+	});
+	subtitleText.anchor.set(0.5, 0.5);
+	subtitleText.x = 1024;
+	subtitleText.y = 950;
+	self.addChild(subtitleText);
+	// Start button
+	var startButton = self.attachAsset('bunker', {
+		anchorX: 0.5,
+		anchorY: 0.5
+	});
+	startButton.x = 1024;
+	startButton.y = 1400;
+	startButton.width = 300;
+	startButton.height = 200;
+	startButton.tint = 0x00AA00;
+	var startText = new Text2('START GAME', {
+		size: 50,
+		fill: 0xFFFFFF
+	});
+	startText.anchor.set(0.5, 0.5);
+	startText.x = 1024;
+	startText.y = 1400;
+	self.addChild(startText);
+	// Instructions text
+	var instructText = new Text2('Use robot towers to stop the alien invasion!\nSave the colonists and defend your colonies!', {
+		size: 40,
+		fill: 0xCCCCCC
+	});
+	instructText.anchor.set(0.5, 0.5);
+	instructText.x = 1024;
+	instructText.y = 1800;
+	self.addChild(instructText);
+	// Add pulsing animation to start button
+	tween(startButton, {
+		scaleX: 1.1,
+		scaleY: 1.1
+	}, {
+		duration: 1000,
+		easing: tween.easeInOut,
+		onFinish: function onFinish() {
+			tween(startButton, {
+				scaleX: 1,
+				scaleY: 1
+			}, {
+				duration: 1000,
+				easing: tween.easeInOut,
+				onFinish: arguments.callee.caller
+			});
+		}
+	});
+	self.down = function (x, y, obj) {
+		// Check if clicked on start button area
+		var dx = x - startButton.x;
+		var dy = y - startButton.y;
+		var distance = Math.sqrt(dx * dx + dy * dy);
+		if (distance <= 150) {
+			startGame();
+		}
+	};
+	return self;
+});
 var Tower = Container.expand(function (towerType) {
 	var self = Container.call(this);
 	self.type = towerType || 'basic';
 	self.range = 150;
@@ -561,8 +647,11 @@
 var bugsToSpawn = 5;
 var bugsSpawned = 0;
 var gameWon = false;
 var towerButtons = [];
+var gameState = 'menu'; // 'menu' or 'playing'
+var mainMenu = null;
+var gameElements = [];
 // Tower costs
 var towerCosts = {
 	basic: 15,
 	rapid: 25,
@@ -623,33 +712,14 @@
 }, {
 	x: 1800,
 	y: 1400
 }];
-// Add background image
-var backgroundImage = game.addChild(LK.getAsset('background', {
-	anchorX: 0.5,
-	anchorY: 0.5
-}));
-backgroundImage.x = 1024;
-backgroundImage.y = 1366;
-backgroundImage.width = 2048;
-backgroundImage.height = 2732;
-// Create colonies array and initial colonies
+// Create colonies array and bunker variable
 var colonies = [];
-var colony1 = game.addChild(LK.getAsset('colony', {
-	anchorX: 0.5,
-	anchorY: 0.5
-}));
-colony1.x = 400;
-colony1.y = 600;
-colonies.push(colony1);
-// Create bunker - make it globally accessible
-var bunker = game.addChild(LK.getAsset('bunker', {
-	anchorX: 0.5,
-	anchorY: 0.5
-}));
-bunker.x = 1800;
-bunker.y = 1400;
+var bunker = null;
+// Initialize main menu
+mainMenu = new MainMenu();
+game.addChild(mainMenu);
 // UI Elements
 var coinText = new Text2('Coins: 50', {
 	size: 60,
 	fill: 0xFFD700
@@ -691,9 +761,95 @@
 	towerButton.y = 2600;
 	game.addChild(towerButton);
 	towerButtons.push(towerButton);
 }
+function startGame() {
+	if (gameState === 'playing') return;
+	gameState = 'playing';
+	// Hide main menu
+	if (mainMenu) {
+		mainMenu.destroy();
+		mainMenu = null;
+	}
+	// Reset game variables
+	bugs = [];
+	towers = [];
+	bullets = [];
+	colonists = [];
+	colonies = [];
+	coins = 80;
+	wave = 1;
+	bugsKilled = 0;
+	colonistsSaved = 0;
+	selectedTower = null;
+	placingTowerType = null;
+	waveTimer = 0;
+	bugsToSpawn = 5;
+	bugsSpawned = 0;
+	gameWon = false;
+	towerButtons = [];
+	// Initialize game elements
+	initializeGameElements();
+	// Start first wave after delay
+	LK.setTimeout(function () {
+		startNewWave();
+	}, 3000);
+}
+function initializeGameElements() {
+	// Add background image
+	var backgroundImage = game.addChild(LK.getAsset('background', {
+		anchorX: 0.5,
+		anchorY: 0.5
+	}));
+	backgroundImage.x = 1024;
+	backgroundImage.y = 1366;
+	backgroundImage.width = 2048;
+	backgroundImage.height = 2732;
+	gameElements.push(backgroundImage);
+	// Create initial colony
+	var colony1 = game.addChild(LK.getAsset('colony', {
+		anchorX: 0.5,
+		anchorY: 0.5
+	}));
+	colony1.x = 400;
+	colony1.y = 600;
+	colonies.push(colony1);
+	gameElements.push(colony1);
+	// Create bunker
+	bunker = game.addChild(LK.getAsset('bunker', {
+		anchorX: 0.5,
+		anchorY: 0.5
+	}));
+	bunker.x = 1800;
+	bunker.y = 1400;
+	gameElements.push(bunker);
+	gameElements.push(towerButton);
+	// Show initial instructions
+	var instructionText = new Text2('Defend the colonies from alien bugs!\nUse robot towers to stop them!', {
+		size: 60,
+		fill: 0xFFFFFF
+	});
+	instructionText.anchor.set(0.5, 0.5);
+	instructionText.x = 1024;
+	instructionText.y = 600;
+	game.addChild(instructionText);
+	gameElements.push(instructionText);
+	tween(instructionText, {
+		alpha: 0
+	}, {
+		duration: 4000,
+		onFinish: function onFinish() {
+			instructionText.destroy();
+		}
+	});
+	// Play background music
+	var bgMusic = LK.getMusic ? LK.getMusic('gameMusic') : null;
+	if (bgMusic || LK.playMusic) {
+		LK.playMusic('gameMusic');
+	}
+}
 function updateUI() {
+	if (gameState !== 'playing') return;
 	coinText.setText('Coins: ' + coins);
 	killsText.setText('Bugs Killed: ' + bugsKilled);
 	savedText.setText('Colonists Saved: ' + colonistsSaved);
 	if (selectedTower) {
@@ -916,8 +1072,9 @@
 		}
 	}
 }
 game.down = function (x, y, obj) {
+	if (gameState !== 'playing') return;
 	// Handle tower upgrade/sell if tower is selected
 	if (selectedTower && x > 1500 && y > 2400) {
 		// Clicked in UI area
 		if (y > 2500) {
@@ -984,8 +1141,9 @@
 	}
 	updateUI();
 };
 game.update = function () {
+	if (gameState !== 'playing') return;
 	if (gameWon) return;
 	// Spawn bugs
 	waveTimer++;
 	if (waveTimer % 45 === 0 && bugsSpawned < bugsToSpawn) {
@@ -1046,31 +1204,5 @@
 	// Update UI periodically
 	if (LK.ticks % 60 === 0) {
 		updateUI();
 	}
-};
-// Show initial instructions
-var instructionText = new Text2('Defend the colonies from alien bugs!\nUse robot towers to stop them!', {
-	size: 60,
-	fill: 0xFFFFFF
-});
-instructionText.anchor.set(0.5, 0.5);
-instructionText.x = 1024;
-instructionText.y = 600;
-game.addChild(instructionText);
-tween(instructionText, {
-	alpha: 0
-}, {
-	duration: 4000,
-	onFinish: function onFinish() {
-		instructionText.destroy();
-	}
-});
-// Start the first wave properly
-LK.setTimeout(function () {
-	startNewWave();
-}, 3000);
-// Play background music
-var bgMusic = LK.getMusic ? LK.getMusic('gameMusic') : null;
-if (bgMusic || LK.playMusic) {
-	LK.playMusic('gameMusic');
-}
\ No newline at end of file
+};
\ No newline at end of file