User input

def question():
    i = 0
    while i < 2:
        answer = input("A = Botnet or B = Hacks:")
        if answer == str("A"):
            print("You have chosen: Botnet")
            break
        elif answer == ('B'):
            print("You have chosen: Hacks")
            break



question()