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

/**** 
* Classes
****/ 
var Hazard = Container.expand(function () {
	var self = Container.call(this);
	var graphics = self.attachAsset('hazard', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.pulseTimer = 0;
	self.update = function () {
		self.pulseTimer += 0.1;
		graphics.scaleX = 1 + Math.sin(self.pulseTimer) * 0.2;
		graphics.scaleY = 1 + Math.sin(self.pulseTimer) * 0.2;
	};
	return self;
});
var Player = Container.expand(function () {
	var self = Container.call(this);
	var graphics = self.attachAsset('microceratus', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.speed = 3;
	self.size = 1;
	self.species = 'microceratus';
	self.dragActive = false;
	self.evolve = function (newSpecies, newSize, newSpeed) {
		// Remove old graphics
		self.removeChild(graphics);
		// Create new graphics based on species
		var assetId = self.getAssetIdForSpecies(newSpecies);
		graphics = self.attachAsset(assetId, {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.species = newSpecies;
		self.size = newSize;
		self.speed = newSpeed;
		graphics.scaleX = newSize;
		graphics.scaleY = newSize;
	};
	self.getAssetIdForSpecies = function (speciesName) {
		var speciesMap = {
			'T-Rex': 'trex',
			'Velociraptor': 'velociraptor',
			'Allosaurus': 'allosaurus',
			'Spinosaurus': 'spinosaurus',
			'Carnotaurus': 'carnotaurus',
			'Dilophosaurus': 'dilophosaurus',
			'Giganotosaurus': 'giganotosaurus',
			'Utahraptor': 'utahraptor',
			'Triceratops': 'triceratops',
			'Brachiosaurus': 'brachiosaurus',
			'Stegosaurus': 'stegosaurus',
			'Ankylosaurus': 'ankylosaurus',
			'Parasaurolophus': 'parasaurolophus',
			'Diplodocus': 'diplodocus',
			'Iguanodon': 'iguanodon',
			'Apatosaurus': 'apatosaurus',
			'Oviraptor': 'oviraptor',
			'Ornithomimus': 'ornithomimus',
			'Therizinosaurus': 'therizinosaurus',
			'Deinonychus': 'deinonychus'
		};
		return speciesMap[speciesName] || 'microceratus';
	};
	return self;
});
var Predator = Container.expand(function (type) {
	var self = Container.call(this);
	var assetName = 'predator' + (type || 1);
	var graphics = self.attachAsset(assetName, {
		anchorX: 0.5,
		anchorY: 0.5
	});
	// Add subtle animation to make predators more lifelike
	graphics.baseScaleX = graphics.scaleX;
	graphics.baseScaleY = graphics.scaleY;
	self.animationTimer = Math.random() * 100;
	self.speed = Math.random() * 2 + 1;
	self.direction = Math.random() * Math.PI * 2;
	self.changeDirectionTimer = 0;
	self.update = function () {
		// Add breathing animation
		self.animationTimer += 0.05;
		graphics.scaleX = graphics.baseScaleX + Math.sin(self.animationTimer) * 0.05;
		graphics.scaleY = graphics.baseScaleY + Math.cos(self.animationTimer * 0.8) * 0.03;
		self.changeDirectionTimer--;
		if (self.changeDirectionTimer <= 0) {
			self.direction = Math.random() * Math.PI * 2;
			self.changeDirectionTimer = 60 + Math.random() * 120;
		}
		self.x += Math.cos(self.direction) * self.speed;
		self.y += Math.sin(self.direction) * self.speed;
		// Bounce off edges
		if (self.x < 100 || self.x > 1948) {
			self.direction = Math.PI - self.direction;
		}
		if (self.y < 100 || self.y > 2632) {
			self.direction = -self.direction;
		}
		// Keep in bounds
		self.x = Math.max(100, Math.min(1948, self.x));
		self.y = Math.max(100, Math.min(2632, self.y));
	};
	return self;
});
var Tree = Container.expand(function (treeType) {
	var self = Container.call(this);
	var treeAsset = 'tree' + (treeType || 1);
	var graphics = self.attachAsset(treeAsset, {
		anchorX: 0.5,
		anchorY: 1.0
	});
	// Random scale variation
	var scale = 0.8 + Math.random() * 0.4;
	graphics.scaleX = scale;
	graphics.scaleY = scale;
	// Subtle swaying animation
	self.swayTimer = Math.random() * 100;
	self.swaySpeed = 0.02 + Math.random() * 0.01;
	self.swayAmount = 0.02 + Math.random() * 0.02;
	self.update = function () {
		self.swayTimer += self.swaySpeed;
		graphics.rotation = Math.sin(self.swayTimer) * self.swayAmount;
	};
	return self;
});

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

/**** 
* Game Code
****/ 
// Game variables
var player;
var predators = [];
var hazards = [];
var trees = [];
var coins = storage.coins || 0;
var survivalTime = 0;
var lastCoinTime = 0;
var gameRunning = true;
var menuOpen = false;
// Keyboard input system
var keys = {
	w: false,
	a: false,
	s: false,
	d: false,
	ArrowUp: false,
	ArrowLeft: false,
	ArrowDown: false,
	ArrowRight: false
};
// Add keyboard event listeners
function initKeyboardControls() {
	if (typeof window !== 'undefined' && window.addEventListener) {
		window.addEventListener('keydown', function (e) {
			if (keys.hasOwnProperty(e.key)) {
				keys[e.key] = true;
				e.preventDefault();
			}
		});
		window.addEventListener('keyup', function (e) {
			if (keys.hasOwnProperty(e.key)) {
				keys[e.key] = false;
				e.preventDefault();
			}
		});
	}
}
// Initialize keyboard controls
initKeyboardControls();
// Dinosaur data
var dinosaurData = {
	carnivores: [{
		name: 'T-Rex',
		cost: 100,
		size: 2.5,
		speed: 4
	}, {
		name: 'Velociraptor',
		cost: 50,
		size: 1.3,
		speed: 6
	}, {
		name: 'Allosaurus',
		cost: 80,
		size: 2.2,
		speed: 3.5
	}, {
		name: 'Spinosaurus',
		cost: 120,
		size: 2.8,
		speed: 3
	}, {
		name: 'Carnotaurus',
		cost: 70,
		size: 2.0,
		speed: 5
	}, {
		name: 'Dilophosaurus',
		cost: 40,
		size: 1.5,
		speed: 4.5
	}, {
		name: 'Giganotosaurus',
		cost: 150,
		size: 3.0,
		speed: 2.5
	}, {
		name: 'Utahraptor',
		cost: 60,
		size: 1.8,
		speed: 5.5
	}],
	herbivores: [{
		name: 'Triceratops',
		cost: 30,
		size: 1.8,
		speed: 2.5
	}, {
		name: 'Brachiosaurus',
		cost: 90,
		size: 3.5,
		speed: 1.5
	}, {
		name: 'Stegosaurus',
		cost: 45,
		size: 2.0,
		speed: 2
	}, {
		name: 'Ankylosaurus',
		cost: 55,
		size: 1.7,
		speed: 1.8
	}, {
		name: 'Parasaurolophus',
		cost: 35,
		size: 1.6,
		speed: 3.5
	}, {
		name: 'Diplodocus',
		cost: 75,
		size: 3.0,
		speed: 2
	}, {
		name: 'Iguanodon',
		cost: 40,
		size: 1.9,
		speed: 3
	}, {
		name: 'Apatosaurus',
		cost: 85,
		size: 3.2,
		speed: 1.8
	}],
	omnivores: [{
		name: 'Oviraptor',
		cost: 25,
		size: 1.2,
		speed: 4
	}, {
		name: 'Ornithomimus',
		cost: 35,
		size: 1.4,
		speed: 5
	}, {
		name: 'Therizinosaurus',
		cost: 65,
		size: 2.3,
		speed: 2.8
	}, {
		name: 'Deinonychus',
		cost: 45,
		size: 1.6,
		speed: 4.8
	}]
};
var unlockedDinosaurs = storage.unlockedDinosaurs || ['microceratus'];
// UI Elements
var coinsText = new Text2('Coins: 0', {
	size: 50,
	fill: '#FFD700'
});
coinsText.anchor.set(1, 0);
LK.gui.topRight.addChild(coinsText);
var timeText = new Text2('Time: 0s', {
	size: 40,
	fill: '#FFFFFF'
});
timeText.anchor.set(0.5, 0);
LK.gui.top.addChild(timeText);
var menuButton = LK.getAsset('menuButton', {
	anchorX: 0.5,
	anchorY: 0.5,
	x: 1900,
	y: 150
});
game.addChild(menuButton);
var menuButtonText = new Text2('EVOLVE', {
	size: 30,
	fill: '#FFFFFF'
});
menuButtonText.anchor.set(0.5, 0.5);
menuButtonText.x = 1900;
menuButtonText.y = 150;
game.addChild(menuButtonText);
// Menu system
var menuContainer;
var menuBackground;
function createEvolutionMenu() {
	menuContainer = new Container();
	menuBackground = LK.getAsset('menuButton', {
		anchorX: 0.5,
		anchorY: 0.5,
		scaleX: 8,
		scaleY: 12,
		x: 1024,
		y: 1366
	});
	menuContainer.addChild(menuBackground);
	var titleText = new Text2('EVOLUTION MENU', {
		size: 50,
		fill: '#FFFFFF'
	});
	titleText.anchor.set(0.5, 0.5);
	titleText.x = 1024;
	titleText.y = 800;
	menuContainer.addChild(titleText);
	var menuCoinsText = new Text2('Coins: ' + coins, {
		size: 45,
		fill: '#FFD700'
	});
	menuCoinsText.anchor.set(0.5, 0.5);
	menuCoinsText.x = 1024;
	menuCoinsText.y = 860;
	menuContainer.addChild(menuCoinsText);
	var closeText = new Text2('TAP ANYWHERE TO CLOSE', {
		size: 30,
		fill: '#CCCCCC'
	});
	closeText.anchor.set(0.5, 0.5);
	closeText.x = 1024;
	closeText.y = 1900;
	menuContainer.addChild(closeText);
	var yPos = 950;
	var categories = ['herbivores', 'carnivores', 'omnivores'];
	categories.forEach(function (category) {
		var categoryText = new Text2(category.toUpperCase(), {
			size: 35,
			fill: '#90EE90'
		});
		categoryText.anchor.set(0.5, 0.5);
		categoryText.x = 1024;
		categoryText.y = yPos;
		menuContainer.addChild(categoryText);
		yPos += 60;
		dinosaurData[category].forEach(function (dino, index) {
			if (unlockedDinosaurs.indexOf(dino.name) !== -1) {
				var dinoText = new Text2(dino.name + ' (OWNED)', {
					size: 35,
					fill: '#00FF00'
				});
			} else if (coins >= dino.cost) {
				var dinoText = new Text2(dino.name + ' - ' + dino.cost + ' coins', {
					size: 35,
					fill: '#FFFFFF'
				});
			} else {
				var dinoText = new Text2(dino.name + ' - ' + dino.cost + ' coins (LOCKED)', {
					size: 35,
					fill: '#888888'
				});
			}
			dinoText.anchor.set(0.5, 0.5);
			dinoText.x = 1024;
			dinoText.y = yPos;
			dinoText.dinoData = dino;
			dinoText.category = category;
			menuContainer.addChild(dinoText);
			yPos += 50;
		});
		yPos += 40;
	});
	game.addChild(menuContainer);
}
function closeEvolutionMenu() {
	if (menuContainer) {
		game.removeChild(menuContainer);
		menuContainer = null;
	}
	menuOpen = false;
	gameRunning = true;
}
// Initialize player
player = new Player();
player.x = 1024;
player.y = 1366;
game.addChild(player);
// Spawn predators
for (var i = 0; i < 4; i++) {
	var predator = new Predator(Math.floor(Math.random() * 3) + 1);
	predator.x = Math.random() * 1848 + 100;
	predator.y = Math.random() * 2532 + 100;
	predators.push(predator);
	game.addChild(predator);
}
// Spawn hazards
for (var i = 0; i < 6; i++) {
	var hazard = new Hazard();
	hazard.x = Math.random() * 1848 + 100;
	hazard.y = Math.random() * 2532 + 100;
	hazards.push(hazard);
	game.addChild(hazard);
}
// Spawn background trees
for (var i = 0; i < 15; i++) {
	var tree = new Tree(Math.floor(Math.random() * 3) + 1);
	tree.x = Math.random() * 1900 + 74;
	tree.y = Math.random() * 2500 + 232;
	trees.push(tree);
	game.addChild(tree);
	// Send trees to back so they appear behind other objects
	game.setChildIndex(tree, 0);
}
function updateCoinsDisplay() {
	coinsText.setText('Coins: ' + coins);
}
function updateTimeDisplay() {
	var seconds = Math.floor(survivalTime / 60);
	timeText.setText('Time: ' + seconds + 's');
}
updateCoinsDisplay();
// Touch handling for dragging
var isDragging = false;
game.down = function (x, y, obj) {
	if (menuOpen) {
		// Check if clicking on a dinosaur in menu
		var localPos = menuContainer.toLocal({
			x: x,
			y: y
		});
		var clickedChild = null;
		for (var i = menuContainer.children.length - 1; i >= 0; i--) {
			var child = menuContainer.children[i];
			if (child.dinoData && localPos.x > child.x - 400 && localPos.x < child.x + 400 && localPos.y > child.y - 30 && localPos.y < child.y + 30) {
				clickedChild = child;
				break;
			}
		}
		if (clickedChild && clickedChild.dinoData) {
			var dino = clickedChild.dinoData;
			if (coins >= dino.cost) {
				// Purchase the dinosaur if not already unlocked
				if (unlockedDinosaurs.indexOf(dino.name) === -1) {
					coins -= dino.cost;
					unlockedDinosaurs.push(dino.name);
					storage.coins = coins;
					storage.unlockedDinosaurs = unlockedDinosaurs;
				}
				// Always evolve player to selected dinosaur
				player.evolve(dino.name, dino.size, dino.speed);
				updateCoinsDisplay();
				LK.getSound('collect').play();
				// Close menu and resume game
				closeEvolutionMenu();
			}
		}
		closeEvolutionMenu();
		return;
	}
	// Check menu button
	var buttonBounds = {
		left: menuButton.x - 100,
		right: menuButton.x + 100,
		top: menuButton.y - 30,
		bottom: menuButton.y + 30
	};
	if (x >= buttonBounds.left && x <= buttonBounds.right && y >= buttonBounds.top && y <= buttonBounds.bottom) {
		menuOpen = true;
		gameRunning = false;
		createEvolutionMenu();
		return;
	}
	isDragging = true;
	player.dragActive = true;
};
game.move = function (x, y, obj) {
	if (isDragging && !menuOpen && player.dragActive) {
		player.x = Math.max(50, Math.min(1998, x));
		player.y = Math.max(50, Math.min(2682, y));
	}
};
game.up = function (x, y, obj) {
	isDragging = false;
	player.dragActive = false;
};
game.update = function () {
	if (!gameRunning) return;
	survivalTime++;
	updateTimeDisplay();
	// Handle keyboard movement
	if (!isDragging && !menuOpen) {
		var moveSpeed = player.speed || 3;
		var newX = player.x;
		var newY = player.y;
		// WASD and Arrow key movement
		if (keys.w || keys.ArrowUp) {
			newY -= moveSpeed;
		}
		if (keys.s || keys.ArrowDown) {
			newY += moveSpeed;
		}
		if (keys.a || keys.ArrowLeft) {
			newX -= moveSpeed;
		}
		if (keys.d || keys.ArrowRight) {
			newX += moveSpeed;
		}
		// Keep player within bounds
		player.x = Math.max(50, Math.min(1998, newX));
		player.y = Math.max(50, Math.min(2682, newY));
	}
	// Coin generation every 5 seconds
	if (survivalTime - lastCoinTime >= 300) {
		// 60 FPS * 5 seconds
		coins += Math.floor(player.size);
		lastCoinTime = survivalTime;
		storage.coins = coins;
		updateCoinsDisplay();
		LK.getSound('collect').play();
	}
	// Check collisions with predators
	for (var i = 0; i < predators.length; i++) {
		var predator = predators[i];
		if (player.intersects(predator)) {
			if (player.size < 2.0) {
				// Small dinosaurs die to predators
				LK.effects.flashScreen(0xFF0000, 1000);
				LK.getSound('danger').play();
				LK.showGameOver();
				return;
			}
		}
	}
	// Check collisions with hazards
	for (var i = 0; i < hazards.length; i++) {
		var hazard = hazards[i];
		if (player.intersects(hazard)) {
			LK.effects.flashScreen(0xFF0000, 500);
			LK.getSound('danger').play();
			LK.showGameOver();
			return;
		}
	}
	// Check win condition (all dinosaurs unlocked)
	var totalDinosaurs = dinosaurData.carnivores.length + dinosaurData.herbivores.length + dinosaurData.omnivores.length + 1;
	if (unlockedDinosaurs.length >= totalDinosaurs) {
		LK.showYouWin();
		return;
	}
};

===================================================================
--- original.js
+++ change.js
@@ -463,24 +463,22 @@
 			}
 		}
 		if (clickedChild && clickedChild.dinoData) {
 			var dino = clickedChild.dinoData;
-			if (unlockedDinosaurs.indexOf(dino.name) === -1 && coins >= dino.cost) {
-				coins -= dino.cost;
-				unlockedDinosaurs.push(dino.name);
-				storage.coins = coins;
-				storage.unlockedDinosaurs = unlockedDinosaurs;
-				// Evolve player
+			if (coins >= dino.cost) {
+				// Purchase the dinosaur if not already unlocked
+				if (unlockedDinosaurs.indexOf(dino.name) === -1) {
+					coins -= dino.cost;
+					unlockedDinosaurs.push(dino.name);
+					storage.coins = coins;
+					storage.unlockedDinosaurs = unlockedDinosaurs;
+				}
+				// Always evolve player to selected dinosaur
 				player.evolve(dino.name, dino.size, dino.speed);
 				updateCoinsDisplay();
 				LK.getSound('collect').play();
-				// Close and reopen menu to refresh coin display
+				// Close menu and resume game
 				closeEvolutionMenu();
-				LK.setTimeout(function () {
-					menuOpen = true;
-					gameRunning = false;
-					createEvolutionMenu();
-				}, 100);
 			}
 		}
 		closeEvolutionMenu();
 		return;