Add logic to check for winner