1 files changed, 8 insertions, 0 deletions
diff --git a/main.py b/main.py
index 43645dc..6081f2a 100644
--- a/main.py
+++ b/main.py
@@ -4,6 +4,7 @@ from circleshape import *
 from player import *
 from asteroid import *
 from asteroidfield import *
+from shot import *
 
 def main():
     pygame.init()
@@ -13,6 +14,7 @@ def main():
     updatable = pygame.sprite.Group()
     drawable = pygame.sprite.Group()
     asteroids = pygame.sprite.Group()
+    shots = pygame.sprite.Group()
 
     Asteroid.containers = (asteroids, updatable, drawable)
     AsteroidField.containers = (updatable)
@@ -22,6 +24,8 @@ def main():
     Player.containers = (updatable, drawable)
     player = Player(SCREEN_WIDTH / 2, SCREEN_HEIGHT / 2)
 
+    Shot.containers = (shots, updatable, drawable)
+    
     dt = 0
 
     while True:
@@ -31,6 +35,10 @@ def main():
         screen.fill("black")
         for obj in updatable:
             obj.update(dt)
+        for asteroid in asteroids:
+            if asteroid.collision_check(player):
+                print("Game over!")
+                return
         for obj in drawable:
             obj.draw(screen)
         pygame.display.flip()