Added update stage for main game loop