construir un programa que muestre el monto que le descontaran a un alumno de su pension, sabie ndo que el porcentaje de descuento es del 20% para los alumnos que han obtenido un promedio entre 16 y 20 de nota

def pension(nota):
    
    if nota >= 16 and nota <= 20:
        return 0.20
    else:
        return 0

def main():
    nota = float(input("ingresa tu nota: "))
    descuento = pension(nota)
    monto = 400*(1 - descuento)
    print(f"El monto a pagar es: {monto}")

main()