mejora con funciones

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Script que implementa una pequeña calculadora de dos nros naturales

def menu():
    return """
    Menu de opciones
    1. Suma
    3. Resta
    3. Multiplicación
    4. División
    5. Salir
    """

def ingresar(num):
    while True:
        n = input(f"Ingrese un número {num}: ")
        if n.isdecimal():
            return int(n)
        else:
            print("Error: debe ingresar un numero entero mayor o igual a cero")

def suma(a,b):
    return f"a + b = {a + b}"

def resta(a,b):
    return f"a - b = {a - b}"

def multiplicacion(a,b):
    return f"a * b = {a * b}"

def division(a,b):
    if b:
        return f"a / b = {a / b}"
    else:
        return "No se puede dividir por cero"


#############    main   ##################################

print("Calculadora de 2 nros naturales")
while True:
    print(menu())

    opcion = input("Seleccione una opción: ")


    if opcion == "1":
        a = ingresar("a")
        b = ingresar("b")
        print(suma(a,b))


    elif opcion == "2":
        a = ingresar("a")
        b = ingresar("b")
        print(resta(a,b))


    elif opcion == "3":
        a = ingresar("a")
        b = ingresar("b")
        print(multiplicacion(a,b))


    elif opcion == "4":
        a = ingresar("a")
        b = ingresar("b")
        print(division(a,b))


    elif opcion == "5":
        print("Gracias por usar este programa...")
        break

    else:
        print("Opción incorrecta")