Minor changes

1 files changed, 6 insertions, 4 deletions

diff --git a/snake.py b/snake.py
index 0445e53..ca21129 100755
--- a/snake.py
+++ b/snake.py
@@ -150,6 +150,11 @@ class Snake:
         # Refresh screen
         self.draw()
         self.clock.tick(self.fps)
+        # Ensure we not quit
+        for event in pygame.event.get():
+                if event.type == pygame.QUIT: 
+                    pygame.quit()
+                    sys.exit()
         return(code)
     
     def play2(self,direction):
@@ -185,10 +190,7 @@ class Snake:
         while True:
             # Check inputs
             for event in pygame.event.get():
-                if event.type == pygame.QUIT: 
-                    pygame.quit()
-                    sys.exit()
-                elif event.type == pygame.KEYDOWN:
+                if event.type == pygame.KEYDOWN:
                     if event.key == pygame.K_LEFT and self.direction != 3:
                         self.direction=9
                         break