Untitled

def get_menu_option():
    while True:
        menu = input("Choose difficulty: ")
        choices = ["1", "2", "3", "4"]
        diffHuVsHu = ""
        diffRAAIVsRAI = ""
        diffHuVsRAI = ""
        diffHUVsUbAI = ""
        if menu in choices:
            if menu == choices[0]:
                difficulty = diffHuVsHu
                return "Selected difficulty is 1"
                break
            elif menu == choices[1]:
                difficulty = diffRAAIVsRAI
                return "Selected difficulty is 2"
                break
            elif menu == choices[2]:
                difficulty = diffHuVsRAI
                return "Selected difficulty is 3"
                break
            elif menu == choices[3]:
                difficulty = diffHUVsUbAI
                return "Selected difficulty is 4"
                break
        else:
            print("Please try numbers between 1 - 4 to select difficulty")