calcular el cobro en una caseta de pago con diferentes clasificaciones de vehículos

def peaje(tipo, cuota):
    if tipo == 'moto':
        return cuota
    elif tipo == 'auto':
        return cuota * 2
    elif tipo == 'camion':
        return cuota * 4
    else:
        return 'Tipo de vehículo incorrecto'

peaje('moto', 20)