Added a main pygame game loop