I am taking a look at a code for an age structured COVID-19 model on GitHub which can be found at this link: https://github.com/p-j-r/covid-19/blob/master/SIR_model_India.py