Added the game loop in main.py