Add check for finding a winner