Binary HW

def binary_add(a, b):
    return bin(int(a, 2) + int(b, 2))[2:]

def binary_subtract(a, b):
    return bin(int(a, 2) - int(b, 2))[2:]

def binary_multiply(a, b):
    return bin(int(a, 2) * int(b, 2))[2:]

def binary_divide(a, b):
    if int(b, 2) == 0:
        return "Error: Division by zero"
    return bin(int(a, 2) // int(b, 2))[2:]

def decimal_to_binary(n):
    return bin(n)[2:]

def main():
    print("Welcome to the Binary Calculator!")
    print("Choose an option:")
    print("1. Add")
    print("2. Subtract")
    print("3. Multiply")
    print("4. Divide")
    print("5. Convert Decimal to Binary (Extra Credit)")

    choice = input("Enter choice (1/2/3/4/5): ")

    if choice in ['1', '2', '3', '4']:
        a = input("Enter first binary number: ")
        b = input("Enter second binary number: ")

        if choice == '1':
            result = binary_add(a, b)
            print("Result (Addition):", result)
        elif choice == '2':
            result = binary_subtract(a, b)
            print("Result (Subtraction):", result)
        elif choice == '3':
            result = binary_multiply(a, b)
            print("Result (Multiplication):", result)
        elif choice == '4':
            result = binary_divide(a, b)
            print("Result (Division):", result)

    elif choice == '5':
        n = int(input("Enter a decimal number: "))
        result = decimal_to_binary(n)
        print("Binary form:", result)

    else:
        print("Invalid choice. Please run the program again.")