Fix win condition and win screen