question = input("Do you want to play again (Y or N)? ")

def f(x):
    if question == "Y":
        return True
    if question == "N":
        return False
    else:
        print("Invalid input")

question = input("Do you want to play again (Y or N)? ")

def f(x):
    if question == "Y":
        return True
    if question == "N":
        return False
    else:
        print("Invalid input")

