diff --git a/main3.py b/main3.py
index cc70c69..c83a0a5 100644
--- a/main3.py
+++ b/main3.py
@@ -40,7 +40,7 @@ def list_music():
 def download(data):
     url = data['youtube_url']
     # ... existing download logic ...
-    emit('download_status', {'status': 'success', 'message': 'Song added successfully'})
+    emit('download_status', {'status': 'success', 'message': 'Song added successfully'}) # Add song ID to response
 
 @socketio.on('play')
 def play(data):
@@ -57,7 +57,7 @@ def get_queue():
 
 @socketio.on('remove_from_queue')
 def remove_from_queue(data):
-    song_id = int(data['songId'])
+    song_id = int(data['songId']) # Use queue index instead
     # ... existing remove logic ...
     emit('remove_status', {'status': 'success', 'message': 'Song removed successfully'})