From fb775a4f225441beaebafbfde2d64c621fd81b7e Mon Sep 17 00:00:00 2001
From: Vlad <89295404+Virashu@users.noreply.github.com>
Date: Tue, 13 Feb 2024 12:57:45 +0300
Subject: [PATCH] Minor changes

---
 danmaku/game.py  | 18 +++++++++++-------
 danmaku/level.py |  0
 2 files changed, 11 insertions(+), 7 deletions(-)
 create mode 100644 danmaku/level.py

diff --git a/danmaku/game.py b/danmaku/game.py
index 36b5238..e13ea14 100644
--- a/danmaku/game.py
+++ b/danmaku/game.py
@@ -229,13 +229,7 @@ def update_game(self):
         self.background_object.animate()
 
         if len(self.enemies) == 0:
-            if len(LEVELS) > self.cur_level + 1:
-                self.cur_level += 1
-                self.enemies = list(LEVELS[self.cur_level])
-            else:
-                set_saved_game(self.cur_level, self.player.score, self.player.power)
-                self.exit_status = "win"
-                self.stop()
+            self.next_level()
 
         if self.player.health <= 0:
             set_saved_game(self.cur_level, self.player.score, self.player.power)
@@ -247,6 +241,16 @@ def update_game(self):
                 pygame.time.wait(10)
             self.stop()
 
+    def next_level(self) -> None:
+        """Start next level if possible"""
+        if len(LEVELS) > self.cur_level + 1:
+            self.cur_level += 1
+            self.enemies = list(LEVELS[self.cur_level])
+        else:
+            set_saved_game(self.cur_level, self.player.score, self.player.power)
+            self.exit_status = "win"
+            self.stop()
+
     def update(self):
         # self.print_stats()
         if Keys.ESCAPE in self.pressed_keys:
diff --git a/danmaku/level.py b/danmaku/level.py
new file mode 100644
index 0000000..e69de29