Added win detection. Still doesn't end if won