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

/**** 
* Classes
****/ 
var Base = Container.expand(function () {
	var self = Container.call(this);
	var baseGraphics = self.attachAsset('base', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.brainrotCount = 0;
	var countText = new Text2('0', {
		size: 60,
		fill: '#ffffff'
	});
	countText.anchor.set(0.5, 0.5);
	self.addChild(countText);
	self.update = function () {
		// Count brainrots in base
		var count = 0;
		for (var i = 0; i < brainrots.length; i++) {
			if (brainrots[i].owner === player && self.intersects(brainrots[i]) && !brainrots[i].isCarried) {
				count++;
			}
		}
		self.brainrotCount = count;
		countText.setText(count.toString());
	};
	return self;
});
var Brainrot = Container.expand(function () {
	var self = Container.call(this);
	self.brainrotGraphics = null;
	self.name = '';
	self.owner = null;
	self.target = null;
	self.isCarried = false;
	self.speed = 2;
	var nameText = new Text2('', {
		size: 30,
		fill: '#ffffff'
	});
	nameText.anchor.set(0.5, 1);
	nameText.y = -40;
	self.addChild(nameText);
	self.setName = function (name) {
		self.name = name;
		nameText.setText(name);
		// Set asset based on name
		var assetMap = {
			'Tralalero Tralala': 'brainrot_tralalero',
			'Capuchino Assasino': 'brainrot_capuchino',
			'Ballerina Capuchina': 'brainrot_ballerina',
			'Lirli Larila': 'brainrot_lirli',
			'Bombardino Crocodrilo': 'brainrot_bombardino',
			'Makakini Bananini': 'brainrot_makakini',
			'Chimpanzini Banaini': 'brainrot_chimpanzini',
			'Orangutini Anassini': 'brainrot_orangutini',
			'Bananita Dolfinita': 'brainrot_bananita'
		};
		var assetName = assetMap[name] || 'brainrot_tralalero';
		if (self.brainrotGraphics) {
			self.removeChild(self.brainrotGraphics);
		}
		self.brainrotGraphics = self.attachAsset(assetName, {
			anchorX: 0.5,
			anchorY: 0.5
		});
		// Move nameText to front
		self.removeChild(nameText);
		self.addChild(nameText);
	};
	self.update = function () {
		if (!self.isCarried && self.target) {
			var dx = self.target.x - self.x;
			var dy = self.target.y - self.y;
			var dist = Math.sqrt(dx * dx + dy * dy);
			if (dist > 10) {
				self.x += dx / dist * self.speed;
				self.y += dy / dist * self.speed;
			}
		}
		// Floating animation
		if (self.brainrotGraphics) {
			self.brainrotGraphics.y = Math.sin(LK.ticks * 0.05) * 5;
		}
	};
	return self;
});
var NPC = Container.expand(function () {
	var self = Container.call(this);
	var npcGraphics = self.attachAsset('npc', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.money = 50;
	self.carriedBrainrot = null;
	self.target = null;
	self.speed = 3;
	self.state = 'idle'; // idle, buying, returning, stealing
	self.stateTimer = 0;
	self.home = null;
	self.update = function () {
		self.stateTimer--;
		if (self.state === 'idle' && self.stateTimer <= 0) {
			// Decide next action
			if (Math.random() < 0.5 && self.money >= 20) {
				self.state = 'buying';
				self.target = shop;
			} else {
				self.state = 'stealing';
				// Find a brainrot to steal
				var potentialTargets = [];
				for (var i = 0; i < brainrots.length; i++) {
					if (brainrots[i].owner && brainrots[i].owner !== self && !brainrots[i].isCarried) {
						potentialTargets.push(brainrots[i]);
					}
				}
				if (potentialTargets.length > 0) {
					self.target = potentialTargets[Math.floor(Math.random() * potentialTargets.length)];
				} else {
					self.state = 'idle';
					self.stateTimer = 60;
				}
			}
		}
		// Move towards target
		if (self.target) {
			var dx = self.target.x - self.x;
			var dy = self.target.y - self.y;
			var dist = Math.sqrt(dx * dx + dy * dy);
			if (dist > 10) {
				self.x += dx / dist * self.speed;
				self.y += dy / dist * self.speed;
			}
			// Check if reached target
			if (dist < 50) {
				if (self.state === 'buying' && self.target === shop) {
					// Buy a brainrot
					if (self.money >= 20) {
						self.money -= 20;
						var newBrainrot = createBrainrot(self.x, self.y);
						newBrainrot.owner = self;
						newBrainrot.target = self.home;
						self.carriedBrainrot = newBrainrot;
						newBrainrot.isCarried = true;
						self.state = 'returning';
						self.target = self.home;
					}
				} else if (self.state === 'stealing' && self.target.parent) {
					// Steal the brainrot
					self.carriedBrainrot = self.target;
					self.carriedBrainrot.isCarried = true;
					self.carriedBrainrot.owner = self;
					self.state = 'returning';
					self.target = self.home;
					LK.getSound('steal').play();
				} else if (self.state === 'returning' && self.target === self.home) {
					// Deposit brainrot
					if (self.carriedBrainrot) {
						self.carriedBrainrot.isCarried = false;
						self.carriedBrainrot.target = self.home;
						self.carriedBrainrot = null;
						LK.getSound('deposit').play();
					}
					self.state = 'idle';
					self.stateTimer = 120;
					self.target = null;
				}
			}
		}
		// Update carried brainrot position
		if (self.carriedBrainrot) {
			self.carriedBrainrot.x = self.x;
			self.carriedBrainrot.y = self.y - 50;
		}
	};
	return self;
});
var NPCBase = Container.expand(function () {
	var self = Container.call(this);
	var baseGraphics = self.attachAsset('npcbase', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.owner = null;
	self.brainrotCount = 0;
	var countText = new Text2('0', {
		size: 60,
		fill: '#ffffff'
	});
	countText.anchor.set(0.5, 0.5);
	self.addChild(countText);
	self.update = function () {
		// Count brainrots in base
		var count = 0;
		for (var i = 0; i < brainrots.length; i++) {
			if (brainrots[i].owner === self.owner && self.intersects(brainrots[i]) && !brainrots[i].isCarried) {
				count++;
			}
		}
		self.brainrotCount = count;
		countText.setText(count.toString());
	};
	return self;
});
var Player = Container.expand(function () {
	var self = Container.call(this);
	var playerGraphics = self.attachAsset('player', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.money = 100;
	self.carriedBrainrot = null;
	self.speed = 6;
	self.velocity = {
		x: 0,
		y: 0
	};
	self.update = function () {
		// Process WASD movement
		self.velocity.x = 0;
		self.velocity.y = 0;
		if (keys.w) self.velocity.y = -self.speed;
		if (keys.s) self.velocity.y = self.speed;
		if (keys.a) self.velocity.x = -self.speed;
		if (keys.d) self.velocity.x = self.speed;
		// Apply velocity to position
		self.x += self.velocity.x;
		self.y += self.velocity.y;
		// Keep player within screen bounds
		self.x = Math.max(40, Math.min(2008, self.x));
		self.y = Math.max(40, Math.min(2692, self.y));
		// Update carried brainrot position
		if (self.carriedBrainrot) {
			self.carriedBrainrot.x = self.x;
			self.carriedBrainrot.y = self.y - 50;
		}
	};
	return self;
});
var Shop = Container.expand(function () {
	var self = Container.call(this);
	var shopGraphics = self.attachAsset('shop', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	return self;
});

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

/**** 
* Game Code
****/ 
// Game variables
// Different colored brainrots for each type
var player;
var playerBase;
var npcs = [];
var npcBases = [];
var brainrots = [];
var shop;
var street;
// Brainrot names
var brainrotNames = ['Tralalero Tralala', 'Capuchino Assasino', 'Ballerina Capuchina', 'Lirli Larila', 'Bombardino Crocodrilo', 'Makakini Bananini', 'Chimpanzini Banaini', 'Orangutini Anassini', 'Bananita Dolfinita'];
// UI Elements
var moneyText = new Text2('Money: $100', {
	size: 60,
	fill: '#00ff00'
});
moneyText.anchor.set(0, 0);
moneyText.x = 150;
LK.gui.topLeft.addChild(moneyText);
var brainrotCountText = new Text2('Brainrots: 0', {
	size: 60,
	fill: '#ffffff'
});
brainrotCountText.anchor.set(0.5, 0);
LK.gui.top.addChild(brainrotCountText);
// Create street in the middle
street = game.addChild(LK.getAsset('street', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 1024,
	y: 1366
}));
// Create player base in corner
playerBase = game.addChild(new Base());
playerBase.x = 200;
playerBase.y = 2532;
// Create shop
shop = game.addChild(new Shop());
shop.x = 1024;
shop.y = 1366;
// Initialize player
player = game.addChild(new Player());
player.x = playerBase.x;
player.y = playerBase.y;
// Create NPCs and their bases
for (var i = 0; i < 3; i++) {
	var npc = game.addChild(new NPC());
	var npcBase = game.addChild(new NPCBase());
	// Position NPC bases in other corners
	if (i === 0) {
		npcBase.x = 1848;
		npcBase.y = 2532;
	} else if (i === 1) {
		npcBase.x = 200;
		npcBase.y = 200;
	} else {
		npcBase.x = 1848;
		npcBase.y = 200;
	}
	npc.x = npcBase.x;
	npc.y = npcBase.y;
	npc.home = npcBase;
	npcBase.owner = npc;
	npcs.push(npc);
	npcBases.push(npcBase);
}
// Helper function to create a brainrot
function createBrainrot(x, y) {
	var brainrot = game.addChild(new Brainrot());
	brainrot.x = x;
	brainrot.y = y;
	brainrot.setName(brainrotNames[Math.floor(Math.random() * brainrotNames.length)]);
	brainrots.push(brainrot);
	return brainrot;
}
// Keyboard controls - WASD movement
var keys = {
	w: false,
	a: false,
	s: false,
	d: false
};
// Note: Since we can't use keyboard events in LK, we'll use a different approach
// Player movement will be controlled by touch/click on screen edges
game.down = function (x, y, obj) {
	// Check if clicking on shop to get a random brainrot
	if (shop.intersects({
		x: x,
		y: y,
		width: 1,
		height: 1
	})) {
		var newBrainrot = createBrainrot(shop.x, shop.y);
		newBrainrot.owner = player;
		newBrainrot.target = playerBase;
		LK.getSound('purchase').play();
	}
	// Check if clicking on brainrot to steal
	for (var i = 0; i < brainrots.length; i++) {
		var brainrot = brainrots[i];
		if (!brainrot.isCarried && brainrot.owner !== player && Math.abs(brainrot.x - x) < 50 && Math.abs(brainrot.y - y) < 50) {
			// Check if player is close enough
			var dist = Math.sqrt(Math.pow(player.x - brainrot.x, 2) + Math.pow(player.y - brainrot.y, 2));
			if (dist < 100) {
				player.carriedBrainrot = brainrot;
				brainrot.isCarried = true;
				brainrot.owner = player;
				LK.getSound('steal').play();
				break;
			}
		}
	}
};
game.up = function (x, y, obj) {
	// Check if releasing brainrot at base
	if (player.carriedBrainrot && player.intersects(playerBase)) {
		player.carriedBrainrot.isCarried = false;
		player.carriedBrainrot.target = playerBase;
		player.carriedBrainrot = null;
		LK.getSound('deposit').play();
	}
};
// Main game update
game.update = function () {
	// Player movement disabled - no keyboard input processing
	// Update brainrot count
	var totalBrainrots = 0;
	for (var i = 0; i < brainrots.length; i++) {
		if (brainrots[i].owner === player && playerBase.intersects(brainrots[i]) && !brainrots[i].isCarried) {
			totalBrainrots++;
		}
	}
	brainrotCountText.setText('Brainrots: ' + totalBrainrots);
	LK.setScore(totalBrainrots);
	// NPCs can steal from player
	for (var i = 0; i < npcs.length; i++) {
		var npc = npcs[i];
		if (!npc.carriedBrainrot && npc.state === 'stealing') {
			// Check if NPC can steal player's carried brainrot
			if (player.carriedBrainrot && npc.intersects(player)) {
				npc.carriedBrainrot = player.carriedBrainrot;
				player.carriedBrainrot.owner = npc;
				player.carriedBrainrot = null;
				npc.state = 'returning';
				npc.target = npc.home;
				LK.getSound('steal').play();
			}
		}
	}
	// Generate money over time
	if (LK.ticks % 180 === 0) {
		// Every 3 seconds
		player.money += 10;
		moneyText.setText('Money: $' + player.money);
		// NPCs also get money
		for (var i = 0; i < npcs.length; i++) {
			npcs[i].money += 10;
		}
	}
	// Win condition
	if (totalBrainrots >= 20) {
		LK.showYouWin();
	}
};
// Create virtual WASD controls
var controlContainer = new Container();
controlContainer.x = 200;
controlContainer.y = 2000;
game.addChild(controlContainer);
// Create control buttons
var buttonSize = 120;
var buttonColor = 0x555555;
var buttonActiveColor = 0x888888;
// W button
var wButton = controlContainer.addChild(LK.getAsset('npc', {
	width: buttonSize,
	height: buttonSize,
	tint: buttonColor,
	anchorX: 0.5,
	anchorY: 0.5,
	x: 0,
	y: -buttonSize
}));
var wText = new Text2('W', {
	size: 60,
	fill: '#ffffff'
});
wText.anchor.set(0.5, 0.5);
wButton.addChild(wText);
// A button
var aButton = controlContainer.addChild(LK.getAsset('npc', {
	width: buttonSize,
	height: buttonSize,
	tint: buttonColor,
	anchorX: 0.5,
	anchorY: 0.5,
	x: -buttonSize,
	y: 0
}));
var aText = new Text2('A', {
	size: 60,
	fill: '#ffffff'
});
aText.anchor.set(0.5, 0.5);
aButton.addChild(aText);
// S button
var sButton = controlContainer.addChild(LK.getAsset('npc', {
	width: buttonSize,
	height: buttonSize,
	tint: buttonColor,
	anchorX: 0.5,
	anchorY: 0.5,
	x: 0,
	y: buttonSize
}));
var sText = new Text2('S', {
	size: 60,
	fill: '#ffffff'
});
sText.anchor.set(0.5, 0.5);
sButton.addChild(sText);
// D button
var dButton = controlContainer.addChild(LK.getAsset('npc', {
	width: buttonSize,
	height: buttonSize,
	tint: buttonColor,
	anchorX: 0.5,
	anchorY: 0.5,
	x: buttonSize,
	y: 0
}));
var dText = new Text2('D', {
	size: 60,
	fill: '#ffffff'
});
dText.anchor.set(0.5, 0.5);
dButton.addChild(dText);
// Button press handlers
wButton.down = function () {
	keys.w = true;
	wButton.tint = buttonActiveColor;
};
wButton.up = function () {
	keys.w = false;
	wButton.tint = buttonColor;
};
aButton.down = function () {
	keys.a = true;
	aButton.tint = buttonActiveColor;
};
aButton.up = function () {
	keys.a = false;
	aButton.tint = buttonColor;
};
sButton.down = function () {
	keys.s = true;
	sButton.tint = buttonActiveColor;
};
sButton.up = function () {
	keys.s = false;
	sButton.tint = buttonColor;
};
dButton.down = function () {
	keys.d = true;
	dButton.tint = buttonActiveColor;
};
dButton.up = function () {
	keys.d = false;
	dButton.tint = buttonColor;
};
// Play background music
LK.playMusic('bgmusic');
;

===================================================================
--- original.js
+++ change.js
@@ -248,14 +248,8 @@
 	var shopGraphics = self.attachAsset('shop', {
 		anchorX: 0.5,
 		anchorY: 0.5
 	});
-	var shopText = new Text2('SHOP', {
-		size: 40,
-		fill: '#000000'
-	});
-	shopText.anchor.set(0.5, 0.5);
-	self.addChild(shopText);
 	return self;
 });
 
 /**** 
@@ -441,9 +435,9 @@
 var buttonSize = 120;
 var buttonColor = 0x555555;
 var buttonActiveColor = 0x888888;
 // W button
-var wButton = controlContainer.addChild(LK.getAsset('shop', {
+var wButton = controlContainer.addChild(LK.getAsset('npc', {
 	width: buttonSize,
 	height: buttonSize,
 	tint: buttonColor,
 	anchorX: 0.5,
@@ -457,9 +451,9 @@
 });
 wText.anchor.set(0.5, 0.5);
 wButton.addChild(wText);
 // A button
-var aButton = controlContainer.addChild(LK.getAsset('shop', {
+var aButton = controlContainer.addChild(LK.getAsset('npc', {
 	width: buttonSize,
 	height: buttonSize,
 	tint: buttonColor,
 	anchorX: 0.5,
@@ -473,9 +467,9 @@
 });
 aText.anchor.set(0.5, 0.5);
 aButton.addChild(aText);
 // S button
-var sButton = controlContainer.addChild(LK.getAsset('shop', {
+var sButton = controlContainer.addChild(LK.getAsset('npc', {
 	width: buttonSize,
 	height: buttonSize,
 	tint: buttonColor,
 	anchorX: 0.5,
@@ -489,9 +483,9 @@
 });
 sText.anchor.set(0.5, 0.5);
 sButton.addChild(sText);
 // D button
-var dButton = controlContainer.addChild(LK.getAsset('shop', {
+var dButton = controlContainer.addChild(LK.getAsset('npc', {
 	width: buttonSize,
 	height: buttonSize,
 	tint: buttonColor,
 	anchorX: 0.5,