adding graphs in the model to represent the box and flux at each timestep