added_proper_exit_to_draw

Author: HSMarieK

lineflow/simulation/line.py

@@ -1,3 +1,4 @@
+import sys
 import simpy
 import pygame
 import numpy as np
@@ -178,6 +179,7 @@ def _draw(self, screen, actions=None):
         for event in pygame.event.get():
             if event.type == pygame.QUIT:
                 teardown_draw()
+                sys.exit()
 
         screen.fill('white')