From fc80ab7ec532879a2711947c7a2ecf61a477f567 Mon Sep 17 00:00:00 2001
From: Kristofers Solo <kristians.cagulis@gmail.com>
Date: Sat, 23 Apr 2022 13:38:12 +0300
Subject: [PATCH] fix global variables

---
 pygame/snake/source/assets/scripts/classes.py | 4 ++--
 pygame/snake/source/assets/scripts/menu.py    | 6 ++++++
 pygame/snake/source/snake.py                  | 4 ++++
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/pygame/snake/source/assets/scripts/classes.py b/pygame/snake/source/assets/scripts/classes.py
index 96ea5c8e..ef30f1af 100644
--- a/pygame/snake/source/assets/scripts/classes.py
+++ b/pygame/snake/source/assets/scripts/classes.py
@@ -90,9 +90,9 @@ class Snake:
 				if index == len(self.body) - 1:
 					self.turns.pop(head_pos)
 			else:
-				from assets.scripts.menu import walls
+				from globals import walls
+				from snake import end_screen
 				if walls:  # end game if goes into the wall
-					from snake import end_screen
 					head.move(head.direction)
 					if head.direction[0] == -1 and head.pos[0] < 0:  # left to right
 						end_screen()
diff --git a/pygame/snake/source/assets/scripts/menu.py b/pygame/snake/source/assets/scripts/menu.py
index a62809c2..f09b0d89 100644
--- a/pygame/snake/source/assets/scripts/menu.py
+++ b/pygame/snake/source/assets/scripts/menu.py
@@ -73,6 +73,9 @@ def options() -> None:
 		for event in pygame.event.get():
 			if event.type == pygame.QUIT:
 				quit()
+			if event.type == pygame.KEYDOWN:
+				if event.key == pygame.K_ESCAPE:
+					main_menu()
 			if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1:
 				if speed_button.check_input(mouse_pos):
 					change_speed()
@@ -101,6 +104,9 @@ def scoreboard() -> None:
 		for event in pygame.event.get():
 			if event.type == pygame.QUIT:
 				quit()
+			if event.type == pygame.KEYDOWN:
+				if event.key == pygame.K_ESCAPE:
+					main_menu()
 			if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1:
 				if back_button.check_input(mouse_pos):
 					main_menu()
diff --git a/pygame/snake/source/snake.py b/pygame/snake/source/snake.py
index 054c3f37..d67c8995 100755
--- a/pygame/snake/source/snake.py
+++ b/pygame/snake/source/snake.py
@@ -47,6 +47,7 @@ def end_screen() -> None:
 
 
 def main() -> None:
+	snakes.clear()
 	from globals import fps, multiplayer, walls
 	pygame.display.set_caption("Snake")
 
@@ -83,6 +84,9 @@ def main() -> None:
 		for event in pygame.event.get():
 			if event.type == pygame.QUIT:
 				quit()
+			if event.type == pygame.KEYDOWN:
+				if event.key == pygame.K_ESCAPE:
+					end_screen()
 
 		for snake in snakes:
 			snake.move()