Add updateStats in game logic