(FIX) Handle game end --> resolves issue #2