I think this block of code should be a function where we would specify the time instants or maybe better tmin, tmax, tstep wdyt?
we should specify the nrow and ncol in the subplots too.