Fix game loop in main.cc a bit