SIR and SEIR model basis added