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

/**** 
* Classes
****/ 
var Bullet = Container.expand(function () {
	var self = Container.call(this);
	var graphics = self.attachAsset('bullet', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.velocityX = 0;
	self.velocityY = 0;
	self.damage = Bullet.prototype.baseDamage || 25;
	self.update = function () {
		self.x += self.velocityX;
		self.y += self.velocityY;
	};
	return self;
});
var Enemy = Container.expand(function () {
	var self = Container.call(this);
	var graphics = self.attachAsset('enemy', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.health = 50;
	self.speed = 1.5;
	self.damage = 20;
	self.waveNumber = currentWave;
	self.update = function () {
		if (player) {
			var angle = Math.atan2(player.y - self.y, player.x - self.x);
			self.x += Math.cos(angle) * self.speed;
			self.y += Math.sin(angle) * self.speed;
		}
	};
	return self;
});
var Player = Container.expand(function () {
	var self = Container.call(this);
	var graphics = self.attachAsset('player', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.health = 100;
	self.maxHealth = 100;
	self.speed = 4;
	self.shootCooldown = 0;
	self.shootRate = 20; // frames between shots
	self.update = function () {
		if (self.shootCooldown > 0) {
			self.shootCooldown--;
		}
		// Auto-shoot at nearest enemy
		if (self.shootCooldown <= 0 && enemies.length > 0) {
			var nearestEnemy = null;
			var nearestDistance = Infinity;
			for (var i = 0; i < enemies.length; i++) {
				var distance = Math.sqrt(Math.pow(enemies[i].x - self.x, 2) + Math.pow(enemies[i].y - self.y, 2));
				if (distance < nearestDistance) {
					nearestDistance = distance;
					nearestEnemy = enemies[i];
				}
			}
			if (nearestEnemy && nearestDistance < 400) {
				var bullet = new Bullet();
				bullet.x = self.x;
				bullet.y = self.y;
				var angle = Math.atan2(nearestEnemy.y - self.y, nearestEnemy.x - self.x);
				bullet.velocityX = Math.cos(angle) * 8;
				bullet.velocityY = Math.sin(angle) * 8;
				bullets.push(bullet);
				game.addChild(bullet);
				self.shootCooldown = self.shootRate;
				LK.getSound('shoot').play();
			}
		}
	};
	return self;
});
var PowerUp = Container.expand(function () {
	var self = Container.call(this);
	var graphics = self.attachAsset('powerup', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.type = 'health'; // health, speed, damage, rate
	return self;
});

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

/**** 
* Game Code
****/ 
// Game variables
var player;
var bullets = [];
var enemies = [];
var powerups = [];
var currentWave = 1;
var enemiesRemaining = 0;
var waveStartDelay = 0;
var gameStarted = false;
var dragTarget = null;
var waveInProgress = false;
var showingPowerUpSelection = false;
var waveTimer = 0;
var waveTimeLimit = 0;
var enemySpawnTimer = 0;
var enemySpawnRate = 0;
var powerUpSelectionUI = null;
// Arena bounds
var arenaLeft = 200;
var arenaRight = 1848;
var arenaTop = 300;
var arenaBottom = 2432;
// UI Elements
var healthText = new Text2('Health: 100', {
	size: 60,
	fill: 0xFF0000
});
healthText.anchor.set(0, 0);
LK.gui.topLeft.addChild(healthText);
healthText.x = 120;
healthText.y = 20;
var waveText = new Text2('Wave: 1', {
	size: 60,
	fill: 0xFFFFFF
});
waveText.anchor.set(0.5, 0);
LK.gui.top.addChild(waveText);
waveText.y = 20;
var scoreText = new Text2('Score: 0', {
	size: 60,
	fill: 0xFFFF00
});
scoreText.anchor.set(1, 0);
LK.gui.topRight.addChild(scoreText);
scoreText.x = -20;
scoreText.y = 20;
var timerText = new Text2('Time: 30', {
	size: 60,
	fill: 0x00FFFF
});
timerText.anchor.set(0.5, 0);
LK.gui.top.addChild(timerText);
timerText.y = 100;
// Initialize player
player = game.addChild(new Player());
player.x = 1024;
player.y = 1366;
// Create power-up selection UI (initially hidden)
function createPowerUpSelectionUI() {
	if (powerUpSelectionUI) {
		powerUpSelectionUI.destroy();
	}
	powerUpSelectionUI = game.addChild(new Container());
	// Background overlay
	var overlay = powerUpSelectionUI.attachAsset('enemy', {
		anchorX: 0.5,
		anchorY: 0.5,
		scaleX: 34,
		scaleY: 45,
		alpha: 0.8
	});
	overlay.x = 1024;
	overlay.y = 1366;
	overlay.tint = 0x000000;
	// Title
	var titleText = new Text2('Choose Power-Up', {
		size: 80,
		fill: 0xFFFFFF
	});
	titleText.anchor.set(0.5, 0.5);
	titleText.x = 1024;
	titleText.y = 800;
	powerUpSelectionUI.addChild(titleText);
	// Power-up options
	var options = [{
		type: 'health',
		name: 'Health +50',
		cost: 50,
		color: 0xFF0000
	}, {
		type: 'speed',
		name: 'Speed +0.5',
		cost: 75,
		color: 0x00FF00
	}, {
		type: 'damage',
		name: 'Damage +10',
		cost: 100,
		color: 0xFFFF00
	}, {
		type: 'rate',
		name: 'Fire Rate +3',
		cost: 125,
		color: 0x00FFFF
	}];
	for (var i = 0; i < options.length; i++) {
		var option = options[i];
		var button = powerUpSelectionUI.addChild(new Container());
		var buttonBg = button.attachAsset('enemy', {
			anchorX: 0.5,
			anchorY: 0.5,
			scaleX: 8,
			scaleY: 2
		});
		buttonBg.tint = option.color;
		var buttonText = new Text2(option.name + ' - ' + option.cost + ' pts', {
			size: 50,
			fill: 0x000000
		});
		buttonText.anchor.set(0.5, 0.5);
		button.addChild(buttonText);
		button.x = 1024;
		button.y = 1000 + i * 150;
		button.powerUpType = option.type;
		button.cost = option.cost;
		button.down = function (x, y, obj) {
			if (LK.getScore() >= obj.parent.cost) {
				applyPowerUp(obj.parent.powerUpType);
				LK.setScore(LK.getScore() - obj.parent.cost);
				scoreText.setText('Score: ' + LK.getScore());
				hidePowerUpSelection();
			}
		};
	}
	// Skip button
	var skipButton = powerUpSelectionUI.addChild(new Container());
	var skipBg = skipButton.attachAsset('enemy', {
		anchorX: 0.5,
		anchorY: 0.5,
		scaleX: 6,
		scaleY: 1.5
	});
	skipBg.tint = 0x666666;
	var skipText = new Text2('Skip', {
		size: 50,
		fill: 0xFFFFFF
	});
	skipText.anchor.set(0.5, 0.5);
	skipButton.addChild(skipText);
	skipButton.x = 1024;
	skipButton.y = 1700;
	skipButton.down = function () {
		hidePowerUpSelection();
	};
	powerUpSelectionUI.alpha = 0;
	tween(powerUpSelectionUI, {
		alpha: 1
	}, {
		duration: 300
	});
}
function showPowerUpSelection() {
	showingPowerUpSelection = true;
	createPowerUpSelectionUI();
}
function hidePowerUpSelection() {
	showingPowerUpSelection = false;
	if (powerUpSelectionUI) {
		tween(powerUpSelectionUI, {
			alpha: 0
		}, {
			duration: 300,
			onFinish: function onFinish() {
				powerUpSelectionUI.destroy();
				powerUpSelectionUI = null;
				startNextWave();
			}
		});
	}
}
function applyPowerUp(type) {
	switch (type) {
		case 'health':
			player.health = Math.min(player.health + 50, player.maxHealth);
			break;
		case 'speed':
			player.speed = Math.min(player.speed + 0.5, 10);
			break;
		case 'damage':
			// Increase bullet damage globally
			Bullet.prototype.baseDamage = (Bullet.prototype.baseDamage || 25) + 10;
			break;
		case 'rate':
			player.shootRate = Math.max(player.shootRate - 3, 3);
			break;
	}
}
function startWave() {
	if (waveInProgress) return;
	waveInProgress = true;
	waveTimer = 0;
	waveTimeLimit = (30 + currentWave * 10) * 60; // 30 + 10*wave seconds at 60fps
	enemySpawnTimer = 0;
	enemySpawnRate = Math.max(120 - currentWave * 5, 30); // Spawn every 2-0.5 seconds
	console.log('Starting wave ' + currentWave + ' for ' + waveTimeLimit / 60 + ' seconds');
}
function startNextWave() {
	waveInProgress = true;
	waveTimer = 0;
	waveTimeLimit = (30 + currentWave * 10) * 60;
	enemySpawnTimer = 0;
	enemySpawnRate = Math.max(120 - currentWave * 5, 30);
	console.log('Starting wave ' + currentWave + ' for ' + waveTimeLimit / 60 + ' seconds');
}
function spawnEnemy() {
	var enemy = new Enemy();
	// Spawn at random edge
	var side = Math.floor(Math.random() * 4);
	switch (side) {
		case 0:
			// top
			enemy.x = arenaLeft + Math.random() * (arenaRight - arenaLeft);
			enemy.y = arenaTop - 50;
			break;
		case 1:
			// right
			enemy.x = arenaRight + 50;
			enemy.y = arenaTop + Math.random() * (arenaBottom - arenaTop);
			break;
		case 2:
			// bottom
			enemy.x = arenaLeft + Math.random() * (arenaRight - arenaLeft);
			enemy.y = arenaBottom + 50;
			break;
		case 3:
			// left
			enemy.x = arenaLeft - 50;
			enemy.y = arenaTop + Math.random() * (arenaBottom - arenaTop);
			break;
	}
	// Scale enemy with wave - bigger and more resistant
	var sizeMultiplier = 1 + (currentWave - 1) * 0.15;
	var healthMultiplier = 1 + (currentWave - 1) * 0.5;
	enemy.children[0].scaleX = sizeMultiplier;
	enemy.children[0].scaleY = sizeMultiplier;
	enemy.health = Math.floor(50 * healthMultiplier);
	enemy.speed = 1.5 + (currentWave - 1) * 0.1;
	enemy.damage = 20 + (currentWave - 1) * 5;
	enemy.waveNumber = currentWave;
	// Tint stronger enemies
	if (currentWave > 5) {
		enemy.children[0].tint = 0xFF0000 + currentWave * 0x110000;
	}
	enemies.push(enemy);
	game.addChild(enemy);
}
function spawnPowerUp(x, y) {
	var powerup = new PowerUp();
	powerup.x = x;
	powerup.y = y;
	var types = ['health', 'speed', 'damage', 'rate'];
	powerup.type = types[Math.floor(Math.random() * types.length)];
	powerups.push(powerup);
	game.addChild(powerup);
}
function collectPowerUp(powerup) {
	switch (powerup.type) {
		case 'health':
			player.health = Math.min(player.health + 30, player.maxHealth);
			break;
		case 'speed':
			player.speed = Math.min(player.speed + 0.5, 8);
			break;
		case 'damage':
			// Increase bullet damage for all future bullets
			break;
		case 'rate':
			player.shootRate = Math.max(player.shootRate - 3, 5);
			break;
	}
	LK.getSound('powerupCollect').play();
}
function handleMove(x, y, obj) {
	if (dragTarget === player) {
		player.x = Math.max(arenaLeft + 40, Math.min(arenaRight - 40, x));
		player.y = Math.max(arenaTop + 40, Math.min(arenaBottom - 40, y));
	}
}
game.move = handleMove;
game.down = function (x, y, obj) {
	dragTarget = player;
	handleMove(x, y, obj);
};
game.up = function (x, y, obj) {
	dragTarget = null;
};
// Start first wave
LK.setTimeout(function () {
	startWave();
	gameStarted = true;
}, 1000);
game.update = function () {
	if (!gameStarted || showingPowerUpSelection) return;
	// Update wave timer and spawn enemies
	if (waveInProgress) {
		waveTimer++;
		enemySpawnTimer++;
		// Spawn enemies at intervals
		if (enemySpawnTimer >= enemySpawnRate) {
			spawnEnemy();
			enemySpawnTimer = 0;
		}
		// Update timer display
		var timeLeft = Math.max(0, Math.ceil((waveTimeLimit - waveTimer) / 60));
		timerText.setText('Time: ' + timeLeft);
		// Check wave completion
		if (waveTimer >= waveTimeLimit) {
			waveInProgress = false;
			// Clear remaining enemies
			for (var k = enemies.length - 1; k >= 0; k--) {
				enemies[k].destroy();
				enemies.splice(k, 1);
			}
			currentWave++;
			console.log('Wave ' + (currentWave - 1) + ' completed! Moving to wave ' + currentWave);
			if (currentWave > 20) {
				LK.showYouWin();
				return;
			}
			waveText.setText('Wave: ' + currentWave);
			player.health = Math.min(player.health + 20, player.maxHealth);
			showPowerUpSelection();
			return;
		}
	}
	// Update bullets
	for (var i = bullets.length - 1; i >= 0; i--) {
		var bullet = bullets[i];
		// Remove bullets that go off screen
		if (bullet.x < 0 || bullet.x > 2048 || bullet.y < 0 || bullet.y > 2732) {
			bullet.destroy();
			bullets.splice(i, 1);
			continue;
		}
		// Check bullet-enemy collisions
		for (var j = enemies.length - 1; j >= 0; j--) {
			var enemy = enemies[j];
			if (bullet.intersects(enemy)) {
				enemy.health -= bullet.damage;
				LK.effects.flashObject(enemy, 0xFFFFFF, 200);
				LK.getSound('enemyHit').play();
				bullet.destroy();
				bullets.splice(i, 1);
				if (enemy.health <= 0) {
					var enemyX = enemy.x;
					var enemyY = enemy.y;
					enemy.destroy();
					enemies.splice(j, 1);
					LK.setScore(LK.getScore() + 10 * enemy.waveNumber);
					scoreText.setText('Score: ' + LK.getScore());
					// 20% chance to drop powerup
					if (Math.random() < 0.2) {
						spawnPowerUp(enemyX, enemyY);
					}
				}
				break;
			}
		}
	}
	// Check player-enemy collisions
	for (var i = enemies.length - 1; i >= 0; i--) {
		var enemy = enemies[i];
		if (player.intersects(enemy)) {
			player.health -= enemy.damage;
			LK.effects.flashObject(player, 0xFF0000, 300);
			enemy.destroy();
			enemies.splice(i, 1);
			if (player.health <= 0) {
				LK.showGameOver();
				return;
			}
		}
	}
	// Check powerup collection
	for (var i = powerups.length - 1; i >= 0; i--) {
		var powerup = powerups[i];
		if (player.intersects(powerup)) {
			collectPowerUp(powerup);
			powerup.destroy();
			powerups.splice(i, 1);
		}
	}
	// Update UI
	healthText.setText('Health: ' + player.health);
};

===================================================================
--- original.js
+++ change.js
@@ -13,9 +13,9 @@
 		anchorY: 0.5
 	});
 	self.velocityX = 0;
 	self.velocityY = 0;
-	self.damage = 25;
+	self.damage = Bullet.prototype.baseDamage || 25;
 	self.update = function () {
 		self.x += self.velocityX;
 		self.y += self.velocityY;
 	};
@@ -29,8 +29,9 @@
 	});
 	self.health = 50;
 	self.speed = 1.5;
 	self.damage = 20;
+	self.waveNumber = currentWave;
 	self.update = function () {
 		if (player) {
 			var angle = Math.atan2(player.y - self.y, player.x - self.x);
 			self.x += Math.cos(angle) * self.speed;
@@ -111,8 +112,14 @@
 var waveStartDelay = 0;
 var gameStarted = false;
 var dragTarget = null;
 var waveInProgress = false;
+var showingPowerUpSelection = false;
+var waveTimer = 0;
+var waveTimeLimit = 0;
+var enemySpawnTimer = 0;
+var enemySpawnRate = 0;
+var powerUpSelectionUI = null;
 // Arena bounds
 var arenaLeft = 200;
 var arenaRight = 1848;
 var arenaTop = 300;
@@ -140,24 +147,176 @@
 scoreText.anchor.set(1, 0);
 LK.gui.topRight.addChild(scoreText);
 scoreText.x = -20;
 scoreText.y = 20;
+var timerText = new Text2('Time: 30', {
+	size: 60,
+	fill: 0x00FFFF
+});
+timerText.anchor.set(0.5, 0);
+LK.gui.top.addChild(timerText);
+timerText.y = 100;
 // Initialize player
 player = game.addChild(new Player());
 player.x = 1024;
 player.y = 1366;
+// Create power-up selection UI (initially hidden)
+function createPowerUpSelectionUI() {
+	if (powerUpSelectionUI) {
+		powerUpSelectionUI.destroy();
+	}
+	powerUpSelectionUI = game.addChild(new Container());
+	// Background overlay
+	var overlay = powerUpSelectionUI.attachAsset('enemy', {
+		anchorX: 0.5,
+		anchorY: 0.5,
+		scaleX: 34,
+		scaleY: 45,
+		alpha: 0.8
+	});
+	overlay.x = 1024;
+	overlay.y = 1366;
+	overlay.tint = 0x000000;
+	// Title
+	var titleText = new Text2('Choose Power-Up', {
+		size: 80,
+		fill: 0xFFFFFF
+	});
+	titleText.anchor.set(0.5, 0.5);
+	titleText.x = 1024;
+	titleText.y = 800;
+	powerUpSelectionUI.addChild(titleText);
+	// Power-up options
+	var options = [{
+		type: 'health',
+		name: 'Health +50',
+		cost: 50,
+		color: 0xFF0000
+	}, {
+		type: 'speed',
+		name: 'Speed +0.5',
+		cost: 75,
+		color: 0x00FF00
+	}, {
+		type: 'damage',
+		name: 'Damage +10',
+		cost: 100,
+		color: 0xFFFF00
+	}, {
+		type: 'rate',
+		name: 'Fire Rate +3',
+		cost: 125,
+		color: 0x00FFFF
+	}];
+	for (var i = 0; i < options.length; i++) {
+		var option = options[i];
+		var button = powerUpSelectionUI.addChild(new Container());
+		var buttonBg = button.attachAsset('enemy', {
+			anchorX: 0.5,
+			anchorY: 0.5,
+			scaleX: 8,
+			scaleY: 2
+		});
+		buttonBg.tint = option.color;
+		var buttonText = new Text2(option.name + ' - ' + option.cost + ' pts', {
+			size: 50,
+			fill: 0x000000
+		});
+		buttonText.anchor.set(0.5, 0.5);
+		button.addChild(buttonText);
+		button.x = 1024;
+		button.y = 1000 + i * 150;
+		button.powerUpType = option.type;
+		button.cost = option.cost;
+		button.down = function (x, y, obj) {
+			if (LK.getScore() >= obj.parent.cost) {
+				applyPowerUp(obj.parent.powerUpType);
+				LK.setScore(LK.getScore() - obj.parent.cost);
+				scoreText.setText('Score: ' + LK.getScore());
+				hidePowerUpSelection();
+			}
+		};
+	}
+	// Skip button
+	var skipButton = powerUpSelectionUI.addChild(new Container());
+	var skipBg = skipButton.attachAsset('enemy', {
+		anchorX: 0.5,
+		anchorY: 0.5,
+		scaleX: 6,
+		scaleY: 1.5
+	});
+	skipBg.tint = 0x666666;
+	var skipText = new Text2('Skip', {
+		size: 50,
+		fill: 0xFFFFFF
+	});
+	skipText.anchor.set(0.5, 0.5);
+	skipButton.addChild(skipText);
+	skipButton.x = 1024;
+	skipButton.y = 1700;
+	skipButton.down = function () {
+		hidePowerUpSelection();
+	};
+	powerUpSelectionUI.alpha = 0;
+	tween(powerUpSelectionUI, {
+		alpha: 1
+	}, {
+		duration: 300
+	});
+}
+function showPowerUpSelection() {
+	showingPowerUpSelection = true;
+	createPowerUpSelectionUI();
+}
+function hidePowerUpSelection() {
+	showingPowerUpSelection = false;
+	if (powerUpSelectionUI) {
+		tween(powerUpSelectionUI, {
+			alpha: 0
+		}, {
+			duration: 300,
+			onFinish: function onFinish() {
+				powerUpSelectionUI.destroy();
+				powerUpSelectionUI = null;
+				startNextWave();
+			}
+		});
+	}
+}
+function applyPowerUp(type) {
+	switch (type) {
+		case 'health':
+			player.health = Math.min(player.health + 50, player.maxHealth);
+			break;
+		case 'speed':
+			player.speed = Math.min(player.speed + 0.5, 10);
+			break;
+		case 'damage':
+			// Increase bullet damage globally
+			Bullet.prototype.baseDamage = (Bullet.prototype.baseDamage || 25) + 10;
+			break;
+		case 'rate':
+			player.shootRate = Math.max(player.shootRate - 3, 3);
+			break;
+	}
+}
 function startWave() {
 	if (waveInProgress) return;
 	waveInProgress = true;
-	var enemiesToSpawn = Math.min(5 + currentWave * 2, 25);
-	enemiesRemaining = enemiesToSpawn;
-	console.log('Starting wave ' + currentWave + ' with ' + enemiesToSpawn + ' enemies');
-	for (var i = 0; i < enemiesToSpawn; i++) {
-		LK.setTimeout(function () {
-			spawnEnemy();
-		}, i * 200);
-	}
+	waveTimer = 0;
+	waveTimeLimit = (30 + currentWave * 10) * 60; // 30 + 10*wave seconds at 60fps
+	enemySpawnTimer = 0;
+	enemySpawnRate = Math.max(120 - currentWave * 5, 30); // Spawn every 2-0.5 seconds
+	console.log('Starting wave ' + currentWave + ' for ' + waveTimeLimit / 60 + ' seconds');
 }
+function startNextWave() {
+	waveInProgress = true;
+	waveTimer = 0;
+	waveTimeLimit = (30 + currentWave * 10) * 60;
+	enemySpawnTimer = 0;
+	enemySpawnRate = Math.max(120 - currentWave * 5, 30);
+	console.log('Starting wave ' + currentWave + ' for ' + waveTimeLimit / 60 + ' seconds');
+}
 function spawnEnemy() {
 	var enemy = new Enemy();
 	// Spawn at random edge
 	var side = Math.floor(Math.random() * 4);
@@ -182,11 +341,21 @@
 			enemy.x = arenaLeft - 50;
 			enemy.y = arenaTop + Math.random() * (arenaBottom - arenaTop);
 			break;
 	}
-	// Scale enemy health with wave
-	enemy.health = 50 + (currentWave - 1) * 15;
+	// Scale enemy with wave - bigger and more resistant
+	var sizeMultiplier = 1 + (currentWave - 1) * 0.15;
+	var healthMultiplier = 1 + (currentWave - 1) * 0.5;
+	enemy.children[0].scaleX = sizeMultiplier;
+	enemy.children[0].scaleY = sizeMultiplier;
+	enemy.health = Math.floor(50 * healthMultiplier);
 	enemy.speed = 1.5 + (currentWave - 1) * 0.1;
+	enemy.damage = 20 + (currentWave - 1) * 5;
+	enemy.waveNumber = currentWave;
+	// Tint stronger enemies
+	if (currentWave > 5) {
+		enemy.children[0].tint = 0xFF0000 + currentWave * 0x110000;
+	}
 	enemies.push(enemy);
 	game.addChild(enemy);
 }
 function spawnPowerUp(x, y) {
@@ -234,9 +403,41 @@
 	startWave();
 	gameStarted = true;
 }, 1000);
 game.update = function () {
-	if (!gameStarted) return;
+	if (!gameStarted || showingPowerUpSelection) return;
+	// Update wave timer and spawn enemies
+	if (waveInProgress) {
+		waveTimer++;
+		enemySpawnTimer++;
+		// Spawn enemies at intervals
+		if (enemySpawnTimer >= enemySpawnRate) {
+			spawnEnemy();
+			enemySpawnTimer = 0;
+		}
+		// Update timer display
+		var timeLeft = Math.max(0, Math.ceil((waveTimeLimit - waveTimer) / 60));
+		timerText.setText('Time: ' + timeLeft);
+		// Check wave completion
+		if (waveTimer >= waveTimeLimit) {
+			waveInProgress = false;
+			// Clear remaining enemies
+			for (var k = enemies.length - 1; k >= 0; k--) {
+				enemies[k].destroy();
+				enemies.splice(k, 1);
+			}
+			currentWave++;
+			console.log('Wave ' + (currentWave - 1) + ' completed! Moving to wave ' + currentWave);
+			if (currentWave > 20) {
+				LK.showYouWin();
+				return;
+			}
+			waveText.setText('Wave: ' + currentWave);
+			player.health = Math.min(player.health + 20, player.maxHealth);
+			showPowerUpSelection();
+			return;
+		}
+	}
 	// Update bullets
 	for (var i = bullets.length - 1; i >= 0; i--) {
 		var bullet = bullets[i];
 		// Remove bullets that go off screen
@@ -258,10 +459,9 @@
 					var enemyX = enemy.x;
 					var enemyY = enemy.y;
 					enemy.destroy();
 					enemies.splice(j, 1);
-					enemiesRemaining--;
-					LK.setScore(LK.getScore() + 10);
+					LK.setScore(LK.getScore() + 10 * enemy.waveNumber);
 					scoreText.setText('Score: ' + LK.getScore());
 					// 20% chance to drop powerup
 					if (Math.random() < 0.2) {
 						spawnPowerUp(enemyX, enemyY);
@@ -278,9 +478,8 @@
 			player.health -= enemy.damage;
 			LK.effects.flashObject(player, 0xFF0000, 300);
 			enemy.destroy();
 			enemies.splice(i, 1);
-			enemiesRemaining--;
 			if (player.health <= 0) {
 				LK.showGameOver();
 				return;
 			}
@@ -296,26 +495,5 @@
 		}
 	}
 	// Update UI
 	healthText.setText('Health: ' + player.health);
-	// Check wave completion - only when all spawned enemies are defeated
-	if (enemies.length === 0 && enemiesRemaining <= 0 && waveStartDelay <= 0) {
-		waveInProgress = false;
-		waveStartDelay = 120; // 2 second delay at 60fps
-		currentWave++;
-		console.log('Wave ' + (currentWave - 1) + ' completed! Starting wave ' + currentWave);
-		if (currentWave > 20) {
-			LK.showYouWin();
-			return;
-		}
-		waveText.setText('Wave: ' + currentWave);
-		// Heal player slightly between waves
-		player.health = Math.min(player.health + 10, player.maxHealth);
-	}
-	// Handle wave start delay
-	if (waveStartDelay > 0) {
-		waveStartDelay--;
-		if (waveStartDelay === 0) {
-			startWave();
-		}
-	}
 };
\ No newline at end of file