Refactor code

main.py

@@ -108,8 +108,8 @@ def main():
         screen.blit(text, text_rect)
         pygame.display.flip()
 
-        while True:
         restart = False
+        while not restart:
             for event in pygame.event.get():
                 if event.type == pglocals.QUIT:
                     logger.info('Received QUIT event')
@@ -119,8 +119,6 @@ def main():
                     if event.key == pglocals.K_r:
                         logger.debug('Restarting game')
                         restart = True
-            if restart:
-                break
 
             clock.tick(5)