def tower_of_hanoi(n, peg1, peg2): if (n == 1): print(f"Move plate 1 to peg {peg2}") else: peg3 = next(iter({"A","B","C"}.difference({peg1, peg2}))) tower_of_hanoi(n-1, peg1, peg3) print(f"Move plate {n} to peg {peg2}" ) tower_of_hanoi(n-1, peg3, peg2) n = int(input("Number of plates : ")) peg_initial = input("We're moving from this peg (A/B/C) : ").upper() peg_final = input("We're moving to this peg (A/B?C) : ").upper() tower_of_hanoi(n, peg_initial, peg_final)