Вывести график по модели SIR, используя библиотеку Matplotlib

import matplotlib.pyplot as pyplot
 
def SIR(population, days, dt):
    susceptible = population - 1
    infected = 1.0
    recovered = 0.0
    recRate = 0.25
    infRate = 0.0004
    SList = [susceptible]
    IList = [infected]
    RList = [recovered]
    timeList =[0]
    for day in range(1, int(days/dt) + 1):
        population = population + (0.0004 * population - 0.25) * dt
        t = day * dt
        timeList.append(t)
        SList.append(SList)
        IList.append(SList)
        RList.append(SList)
 
    pyplot.plot(timeList, SList, label = 'Susceptible')
    pyplot.plot(timeList, IList, label = 'Infected')
    pyplot.plot(timeList, RList, label = 'Recovered')
    pyplot.legend(loc = 'center right')
    pyplot.xlabel('Days')
    pyplot.ylabel('Individuals')
    pyplot.show()
 
SIR(2200, 30, 0.01)

import matplotlib.pyplot as pyplot
 
def SIR(population, days, dt):
    susceptible = population - 1
    infected = 1.0
    recovered = 0.0
    recRate = 0.25
    infRate = 0.0004
    SList = [susceptible]
    IList = [infected]
    RList = [recovered]
    timeList =[0]
    for day in range(1, int(days/dt) + 1):
        population = population + (0.0004 * population - 0.25) * dt
        t = day * dt
        timeList.append(t)
        SList.append(SList)
        IList.append(SList)
        RList.append(SList)
 
    pyplot.plot(timeList, SList, label = 'Susceptible')
    pyplot.plot(timeList, IList, label = 'Infected')
    pyplot.plot(timeList, RList, label = 'Recovered')
    pyplot.legend(loc = 'center right')
    pyplot.xlabel('Days')
    pyplot.ylabel('Individuals')
    pyplot.show()
 
SIR(2200, 30, 0.01)