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

/**** 
* Classes
****/ 
// Game starts in menu state
var MenuButton = Container.expand(function (text, onClickCallback) {
	var self = Container.call(this);
	// Create button background
	var buttonBg = self.attachAsset('notePad', {
		anchorX: 0.5,
		anchorY: 0.5,
		scaleX: 3.0,
		scaleY: 2.4
	});
	buttonBg.tint = 0x00ff00;
	// Create button text
	var buttonText = new Text2(text, {
		size: 120,
		fill: 0x000000
	});
	buttonText.anchor.set(0.5, 0.5);
	self.addChild(buttonText);
	self.onClickCallback = onClickCallback;
	self.down = function (x, y, obj) {
		buttonBg.tint = 0x00cc00;
		if (self.onClickCallback) {
			self.onClickCallback();
		}
	};
	self.up = function (x, y, obj) {
		buttonBg.tint = 0x00ff00;
	};
	return self;
});
var Note = Container.expand(function (lane) {
	var self = Container.call(this);
	var noteGraphics = self.attachAsset('note', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.lane = lane; // 0 = left, 1 = center, 2 = right
	self.speed = Note.prototype.speed || 8;
	self.hasBeenHit = false;
	self.update = function () {
		self.y += self.speed;
	};
	return self;
});
var NotePad = Container.expand(function (lane) {
	var self = Container.call(this);
	var padGraphics = self.attachAsset('notePad', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	var perfectZone = self.attachAsset('perfectZone', {
		anchorX: 0.5,
		anchorY: 0.5,
		alpha: 0.3
	});
	self.lane = lane;
	self.isPressed = false;
	self.down = function (x, y, obj) {
		self.isPressed = true;
		padGraphics.tint = 0xffffff;
		checkNoteHit(self.lane);
	};
	self.up = function (x, y, obj) {
		self.isPressed = false;
		padGraphics.tint = 0x4a4a4a;
	};
	return self;
});

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

/**** 
* Game Code
****/ 
// Create background
var background = LK.getAsset('Background', {
	anchorX: 0,
	anchorY: 0,
	x: 0,
	y: 0
});
game.addChild(background);
var notes = [];
var notePads = [];
var gameScore = 0;
var gameCombo = 0;
var spawnTimer = 0;
var spawnInterval = 45; // 0.75 seconds at 60fps
var perfectZoneY = 2400;
var perfectTolerance = 100;
// Game state
var gameState = 'menu'; // 'menu', 'difficulty', or 'playing'
var selectedDifficulty = 'normal';
var mainMenu = new Container();
var difficultyMenu = new Container();
var gameContainer = new Container();
// Main Menu Elements
var titleText = new Text2('Rhythm Tap Challenge', {
	size: 80,
	fill: 0xFFFFFF
});
titleText.anchor.set(0.5, 0.5);
titleText.x = 1024;
titleText.y = 800;
mainMenu.addChild(titleText);
var startButton = new MenuButton('START', function () {
	showDifficultyMenu();
});
startButton.x = 1024;
startButton.y = 1200;
mainMenu.addChild(startButton);
// Difficulty Menu Elements
var difficultyTitle = new Text2('Select Difficulty', {
	size: 70,
	fill: 0xFFFFFF
});
difficultyTitle.anchor.set(0.5, 0.5);
difficultyTitle.x = 1024;
difficultyTitle.y = 800;
difficultyMenu.addChild(difficultyTitle);
var easyButton = new MenuButton('EASY', function () {
	selectedDifficulty = 'easy';
	startGame();
});
easyButton.x = 1024;
easyButton.y = 1000;
difficultyMenu.addChild(easyButton);
var normalButton = new MenuButton('NORMAL', function () {
	selectedDifficulty = 'normal';
	startGame();
});
normalButton.x = 1024;
normalButton.y = 1150;
difficultyMenu.addChild(normalButton);
var hardButton = new MenuButton('HARD', function () {
	selectedDifficulty = 'hard';
	startGame();
});
hardButton.x = 1024;
hardButton.y = 1300;
difficultyMenu.addChild(hardButton);
var backButton = new MenuButton('BACK', function () {
	showMainMenu();
});
backButton.x = 1024;
backButton.y = 1500;
difficultyMenu.addChild(backButton);
game.addChild(mainMenu);
game.addChild(difficultyMenu);
difficultyMenu.visible = false;
// UI Elements
var scoreText = new Text2('Score: 0', {
	size: 60,
	fill: 0xFFFFFF
});
scoreText.anchor.set(1, 0);
scoreText.x = -50;
scoreText.y = 50;
LK.gui.topRight.addChild(scoreText);
var comboText = new Text2('Combo: 0', {
	size: 50,
	fill: 0xFFFF00
});
comboText.anchor.set(1, 0);
comboText.x = -50;
comboText.y = 130;
LK.gui.topRight.addChild(comboText);
var highScoreText = new Text2('High Score: ' + (storage.highScore || 0), {
	size: 40,
	fill: 0xFFFFFF
});
highScoreText.anchor.set(1, 0);
highScoreText.x = -50;
highScoreText.y = 200;
LK.gui.topRight.addChild(highScoreText);
// Hide UI initially
scoreText.visible = false;
comboText.visible = false;
highScoreText.visible = false;
// Create restart button for game screen
var restartButton = new MenuButton('↺', function () {
	// Reset game state
	gameScore = 0;
	gameCombo = 0;
	spawnTimer = 0;
	// Clear all notes
	for (var i = notes.length - 1; i >= 0; i--) {
		removeNote(notes[i]);
	}
	updateUI();
});
restartButton.x = 924;
restartButton.y = 120;
restartButton.scaleX = 0.42;
restartButton.scaleY = 0.42;
gameContainer.addChild(restartButton);
// Create return to main menu button for game screen
var returnMenuButton = new MenuButton('🏠', function () {
	// Stop game music and start menu music
	LK.stopMusic();
	LK.playMusic('menumusic', {
		loop: true
	});
	gameState = 'menu';
	gameContainer.visible = false;
	scoreText.visible = false;
	comboText.visible = false;
	highScoreText.visible = false;
	// Clear all notes
	for (var i = notes.length - 1; i >= 0; i--) {
		removeNote(notes[i]);
	}
	// Reset game variables
	gameScore = 0;
	gameCombo = 0;
	spawnTimer = 0;
	updateUI();
	// Show main menu
	showMainMenu();
});
returnMenuButton.x = 1124;
returnMenuButton.y = 120;
returnMenuButton.scaleX = 0.42;
returnMenuButton.scaleY = 0.42;
gameContainer.addChild(returnMenuButton);
// Create note pads
var padPositions = [512, 1024, 1536]; // Left, center, right positions
for (var i = 0; i < 3; i++) {
	var pad = new NotePad(i);
	pad.x = padPositions[i];
	pad.y = perfectZoneY;
	notePads.push(pad);
	gameContainer.addChild(pad);
}
game.addChild(gameContainer);
gameContainer.visible = false;
// Start menu music
LK.playMusic('menumusic', {
	loop: true
});
function showDifficultyMenu() {
	gameState = 'difficulty';
	mainMenu.visible = false;
	difficultyMenu.visible = true;
}
function showMainMenu() {
	gameState = 'menu';
	mainMenu.visible = true;
	difficultyMenu.visible = false;
}
function startGame() {
	gameState = 'playing';
	mainMenu.visible = false;
	difficultyMenu.visible = false;
	gameContainer.visible = true;
	scoreText.visible = true;
	comboText.visible = true;
	highScoreText.visible = true;
	// Reset game state
	gameScore = 0;
	gameCombo = 0;
	spawnTimer = 0;
	// Set note speed based on difficulty
	var noteSpeed = 8; // normal speed
	if (selectedDifficulty === 'easy') {
		noteSpeed = 6;
	} else if (selectedDifficulty === 'hard') {
		noteSpeed = 12;
	}
	// Update all note pads with new speed
	Note.prototype.speed = noteSpeed;
	// Clear any existing notes
	for (var i = notes.length - 1; i >= 0; i--) {
		removeNote(notes[i]);
	}
	// Stop menu music and start game music
	LK.stopMusic();
	LK.playMusic('bgmusic', {
		loop: true
	});
	updateUI();
}
function spawnNote() {
	var randomLane = Math.floor(Math.random() * 3);
	var note = new Note(randomLane);
	note.x = padPositions[randomLane];
	note.y = -50;
	notes.push(note);
	gameContainer.addChild(note);
}
function checkNoteHit(lane) {
	var hitNote = null;
	var bestDistance = Infinity;
	// Find the closest note in the perfect zone for this lane
	for (var i = 0; i < notes.length; i++) {
		var note = notes[i];
		if (note.lane === lane && !note.hasBeenHit) {
			var distance = Math.abs(note.y - perfectZoneY);
			if (distance < perfectTolerance && distance < bestDistance) {
				hitNote = note;
				bestDistance = distance;
			}
		}
	}
	if (hitNote) {
		// Perfect hit
		hitNote.hasBeenHit = true;
		gameScore += 5;
		gameCombo += 1;
		// Combo bonus
		if (gameCombo >= 3) {
			gameScore += 5;
			showComboEffect(hitNote.x, hitNote.y);
		}
		// Visual effect
		tween(hitNote, {
			alpha: 0,
			scaleX: 2,
			scaleY: 2
		}, {
			duration: 200,
			onFinish: function onFinish() {
				removeNote(hitNote);
			}
		});
		LK.getSound('success').play();
		updateUI();
	} else {
		// Wrong timing
		gameScore -= 5;
		gameCombo = 0;
		LK.effects.flashScreen(0xff0000, 200);
		updateUI();
	}
}
function showComboEffect(x, y) {
	var comboEffect = new Text2('COMBO!', {
		size: 80,
		fill: 0xFFFF00
	});
	comboEffect.anchor.set(0.5, 0.5);
	comboEffect.x = x;
	comboEffect.y = y;
	gameContainer.addChild(comboEffect);
	tween(comboEffect, {
		y: y - 100,
		alpha: 0,
		scaleX: 1.5,
		scaleY: 1.5
	}, {
		duration: 1000,
		easing: tween.easeOut,
		onFinish: function onFinish() {
			gameContainer.removeChild(comboEffect);
		}
	});
}
function removeNote(note) {
	var index = notes.indexOf(note);
	if (index > -1) {
		notes.splice(index, 1);
		gameContainer.removeChild(note);
	}
}
function updateUI() {
	scoreText.setText('Score: ' + gameScore);
	comboText.setText('Combo: ' + gameCombo);
	LK.setScore(gameScore);
	// Update high score if current score is higher
	var currentHighScore = storage.highScore || 0;
	if (gameScore > currentHighScore) {
		storage.highScore = gameScore;
		highScoreText.setText('High Score: ' + gameScore);
	}
	// Check for game over at -50 points
	if (gameScore <= -50) {
		LK.showGameOver();
	}
}
// Add pause menu functionality
var pauseMenu = new Container();
var pauseTitle = new Text2('Game Paused', {
	size: 70,
	fill: 0xFFFFFF
});
pauseTitle.anchor.set(0.5, 0.5);
pauseTitle.x = 1024;
pauseTitle.y = 1000;
pauseMenu.addChild(pauseTitle);
var resumeButton = new MenuButton('RESUME', function () {
	pauseMenu.visible = false;
});
resumeButton.x = 1024;
resumeButton.y = 1200;
pauseMenu.addChild(resumeButton);
var mainMenuButton = new MenuButton('MAIN MENU', function () {
	// Stop game music and start menu music
	LK.stopMusic();
	LK.playMusic('menumusic', {
		loop: true
	});
	gameState = 'menu';
	pauseMenu.visible = false;
	gameContainer.visible = false;
	scoreText.visible = false;
	comboText.visible = false;
	highScoreText.visible = false;
	// Clear all notes
	for (var i = notes.length - 1; i >= 0; i--) {
		removeNote(notes[i]);
	}
	// Reset game variables
	gameScore = 0;
	gameCombo = 0;
	spawnTimer = 0;
	updateUI();
	// Show main menu
	showMainMenu();
});
mainMenuButton.x = 1024;
mainMenuButton.y = 1350;
pauseMenu.addChild(mainMenuButton);
game.addChild(pauseMenu);
pauseMenu.visible = false;
game.update = function () {
	if (gameState !== 'playing') {
		return;
	}
	// Spawn notes
	spawnTimer++;
	if (spawnTimer >= spawnInterval) {
		spawnNote();
		spawnTimer = 0;
	}
	// Update and check notes
	for (var i = notes.length - 1; i >= 0; i--) {
		var note = notes[i];
		// Check if note reached bottom (missed)
		if (note.y > 2800 && !note.hasBeenHit) {
			gameScore -= 5;
			gameCombo = 0;
			LK.getSound('fail').play();
			updateUI();
			removeNote(note);
		}
	}
};

===================================================================
--- original.js
+++ change.js
@@ -191,9 +191,9 @@
 scoreText.visible = false;
 comboText.visible = false;
 highScoreText.visible = false;
 // Create restart button for game screen
-var restartButton = new MenuButton('RESTART', function () {
+var restartButton = new MenuButton('↺', function () {
 	// Reset game state
 	gameScore = 0;
 	gameCombo = 0;
 	spawnTimer = 0;