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