Improve game loop