/**** 
* Plugins
****/ 
var tween = LK.import("@upit/tween.v1");
var storage = LK.import("@upit/storage.v1", {
	highScore: 0
});

/**** 
* Classes
****/ 
/**** 
* Balloon Class
****/ 
var Balloon = Container.expand(function (type) {
	var self = Container.call(this);
	var balloonTypes = {
		'red': {
			color: 'balloon_red',
			points: 10,
			speed: 2
		},
		'blue': {
			color: 'balloon_blue',
			points: 20,
			speed: 3
		},
		'yellow': {
			color: 'balloon_yellow',
			points: 30,
			speed: 4
		},
		'green': {
			color: 'balloon_green',
			points: 50,
			speed: 5
		},
		'spiky': {
			color: 'balloon_spiky',
			points: -20,
			speed: 3
		},
		// Spiky balloon decreases points
		'gold': {
			color: 'balloon_gold',
			points: 100,
			speed: 6
		} // Golden balloon is rare & gives high points
	};
	var balloonType = balloonTypes[type] || balloonTypes['red'];
	self.points = balloonType.points;
	self.speed = balloonType.speed;
	self.active = true;
	// Balloon Body
	var balloonBody = self.attachAsset(balloonType.color, {
		anchorX: 0.5,
		anchorY: 0.5
	});
	// Balloon String
	var balloonString = self.attachAsset('balloon_string', {
		anchorX: 0.5,
		anchorY: 0,
		y: balloonBody.height / 2
	});
	// Movement
	self.update = function () {
		if (self.active) {
			self.y -= self.speed;
			if (self.y < -50) {
				self.active = false;
				if (type !== 'spiky' && type !== 'gold') {
					health--; // Lose health if a regular balloon escapes
					LK.getSound('damage').play();
					updateHealthUI();
					if (health <= 0) {
						gameOver();
					}
				}
			}
		}
	};
	// Pop Function
	self.pop = function () {
		if (self.active) {
			self.active = false;
			LK.getSound('pop').play();
			tween(balloonBody, {
				scaleX: 1.5,
				scaleY: 1.5,
				alpha: 0
			}, {
				duration: 300
			});
			score += self.points;
			updateScore();
		}
	};
	return self;
});
/**** 
* Power-Up Class
****/ 
var PowerUp = Container.expand(function (type) {
	var self = Container.call(this);
	self.type = type;
	self.active = true;
	var powerupGraphics = self.attachAsset('powerup', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	// Movement
	self.update = function () {
		if (self.active) {
			self.y -= 2;
			if (self.y < -50) {
				self.active = false;
			}
		}
	};
	// Collect
	self.collect = function () {
		if (self.active) {
			self.active = false;
			LK.getSound('powerup').play();
			if (self.type === 'health') {
				health++;
				updateHealthUI();
			}
		}
	};
	return self;
});

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

/**** 
* Game Code
****/ 
/**** 
* Game Variables
****/ 
var score = 0;
var health = 3;
var gameActive = true;
var waterGun = game.addChild(new Container());
waterGun.x = 1024;
waterGun.y = 2700;
/**** 
* UI Elements
****/ 
var scoreTxt = new Text2('Score: 0', {
	size: 50,
	fill: 0xffffff
});
var healthTxt = new Text2('Health: 3', {
	size: 50,
	fill: 0xff0000
});
LK.gui.topRight.addChild(scoreTxt);
LK.gui.topLeft.addChild(healthTxt);
/**** 
* Functions
****/ 
function spawnBalloon() {
	var types = ['red', 'blue', 'yellow', 'green', 'spiky', 'gold'];
	var balloon = new Balloon(types[Math.floor(Math.random() * types.length)]);
	balloon.x = 200 + Math.random() * 1600;
	balloon.y = 2800;
	game.addChild(balloon);
	LK.setTimeout(spawnBalloon, 1000);
}
function spawnPowerUp() {
	var powerup = new PowerUp('health');
	powerup.x = 200 + Math.random() * 1600;
	powerup.y = 2800;
	game.addChild(powerup);
	LK.setTimeout(spawnPowerUp, 20000);
}
function updateScore() {
	scoreTxt.text = 'Score: ' + score;
	if (score > storage.highScore) {
		storage.highScore = score;
	}
}
function updateHealthUI() {
	healthTxt.text = 'Health: ' + health;
}
function gameOver() {
	gameActive = false;
	LK.gui.center.addChild(new Text2('Game Over!', {
		size: 100,
		fill: 0xff0000
	}));
	LK.setTimeout(function () {
		LK.showGameOver();
	}, 3000);
}
/**** 
* Shooting Mechanism
****/ 
function shoot() {
	if (!gameActive) {
		return;
	}
	var bullet = game.addChild(new Container());
	bullet.x = waterGun.x;
	bullet.y = waterGun.y - 50;
	LK.getSound('shoot').play();
	bullet.update = function () {
		bullet.y -= 20;
		if (bullet.y < -50) {
			bullet.active = false;
		}
	};
	LK.setTimeout(shoot, 500);
}
/**** 
* Start Game
****/ 
spawnBalloon();
spawnPowerUp();
shoot();

===================================================================
--- original.js
+++ change.js
@@ -199,9 +199,9 @@
 		size: 100,
 		fill: 0xff0000
 	}));
 	LK.setTimeout(function () {
-		LK.reload();
+		LK.showGameOver();
 	}, 3000);
 }
 /**** 
 * Shooting Mechanism