Added Pygame main loop