started on main menu system, 3 gamemodes.

main.py

@@ -8,6 +8,22 @@ screen = pygame.display.set_mode((WIDTH, HEIGHT))
 pygame.display.set_caption("Sugarmania")
 clock = pygame.time.Clock()
 font = pygame.font.SysFont(None, 36)
+big_font = pygame.font.SysFont(None, 64)
+small_font = pygame.font.SysFont(None, 26)
+
+MENU_OPTIONS = ["Play", "Quit"]
+
+MODE_NAMES = ["easy", "medium", "hard"]
+
+MODES = {
+    "easy": {"lives": 5, "speed": 7, "spawn_every": 10, "enemy_pool": ["Bouncer"], "desc": "Only Bouncers. One new enemy per 10 sugar"},
+    "medium": {"lives": 3, "speed": 6, "spawn_every": 7, "enemy_pool": ["Bouncer", "Chaser"], "desc": "Bouncers and Chasers. One new enemy per 7 sugar"},
+    "hard": {"lives": 2, "speed": 5, "spawn_every": 5, "enemy_pool": ["Bouncer", "Chaser", "Dasher"], "desc": "Bouncers, Chasers and Dashers. One new enemy per 5 sugar"}
+}
+
+
+
+
 
 class ChaserEnemy:
     def __init__(self, avoid_rect):
@@ -50,7 +66,7 @@ class DasherEnemy:
         if self.state == "wait":
             dx = target.centerx - self.rect.centerx
             dy = target.centery - self.rect.centery
-            dist = max(1, (dx ** 2 + dy ** 2) ** 0,5)
+            dist = max(1, (dx ** 2 + dy ** 2) ** 0.5)
             self.vx = dx / dist * 9
             self.vy = dy / dist * 9
             self.state = "dash"
@@ -77,7 +93,7 @@ class BouncerEnemy:
         self.vy = random.choice([-3, -2, 2, 3])
         self.spawn_timer = 60
 
-    def update(self):
+    def update(self, target=None):
         self.rect.x += self.vx
         self.rect.y += self.vy
         if self.rect.left < 0 or self.rect.right > WIDTH:
@@ -86,23 +102,13 @@ class BouncerEnemy:
             self.vy = -self.vy
 
 
-def enemy_for_level(level): #idk if I'll add the levels
-        if level == 2:
-            return ChaserEnemy
-        if level == 3:
-            return DasherEnemy
-        if level == 4:
-            return DasherEnemy
-        return random.choice([BouncerEnemy, DasherEnemy, ChaserEnemy])
-
-
+ENEMY_CLASSES = {"Bouncer": BouncerEnemy, "Chaser": ChaserEnemy, "Dasher": DasherEnemy}
 
 
 
 player_img = pygame.image.load("player.png").convert_alpha()
 player_img = pygame.transform.scale(player_img, (40, 60))
-player = pygame.Rect(300, 220, 40, 60)
-speed = 5
+
 
 Bouncer_img = pygame.image.load("Bouncer.png").convert_alpha()
 Bouncer_img = pygame.transform.scale(Bouncer_img, (30, 45))
@@ -116,8 +122,6 @@ Dasher_img = pygame.transform.scale(Dasher_img, (30, 45))
 
 sugar_img = pygame.image.load("honey.png").convert_alpha()
 sugar_img = pygame.transform.scale(sugar_img, (50, 50))
-score = 0
-Enemies = []
 
 
 heart_img = pygame.image.load("Heart.png").convert_alpha()
@@ -127,19 +131,27 @@ heart_img = pygame.transform.scale(heart_img,(40, 40))
 def new_sugar():
     return pygame.Rect(random.randint(0, WIDTH - 20),
                        random.randint(0, HEIGHT - 20), 50, 50) #20,s
-sugar = new_sugar()
-lives = 3
-hit_timer = 0
+
 
 
 
 
 #game loop
-running = True
-while running:
+def game_screen(mode):
+    cfg = MODES[mode]
+    player = pygame.Rect(300, 220, 40, 60)
+    speed = cfg["speed"]
+    score = 0
+    Enemies = []
+    sugar = new_sugar()
+    lives = cfg["lives"]
+    hit_timer = 0
+
+    while True:
       for event in pygame.event.get():
         if event.type == pygame.QUIT:
-            running = False
+            return "quit", score
+
 
         #movement
         keys = pygame.key.get_pressed()
@@ -156,7 +168,7 @@ while running:
            if enemy.spawn_timer > 0:
             enemy.spawn_timer -= 1
            else:
-        enemy.update()
+            enemy.update(player)
         if hit_timer > 0:
             hit_timer -= 1
         else:
@@ -166,7 +178,7 @@ while running:
                     hit_timer = 90
                     break
         if lives <= 0:
-        running = False
+            return "gameover", score
 
         #keep player in
         player.clamp_ip(screen.get_rect())
@@ -175,8 +187,9 @@ while running:
         if player.colliderect(sugar):
             score += 1
             sugar = new_sugar()
-        if score % 10 == 0:
-            Enemies.append(BouncerEnemy(player))
+            if score % cfg["spawn_every"] == 0 and cfg["enemy_pool"]:
+                enemy_name = random.choice(cfg["enemy_pool"])
+                Enemies.append(ENEMY_CLASSES[enemy_name](player))
 
 
         screen.fill((255, 228, 240))  # screen
@@ -201,3 +214,7 @@ while running:
 
 pygame.quit()
 
+def menu_screen():
+    selected = 0
+    while True:
+        for