Something like this, but as a class method. Or bundled as part of the `plot_time_model` method? 
Something like this, but as a class method. Or bundled as part of the
plot_time_modelmethod?