Changes:

1. Dates are pulled out of CovidUKStochastic
2. CovidUKStochastic now behaves more like a tfd.Distribution
    * CovidUKStochastic is now instantiated with an initial time, number of time steps and
time step size
    * CovidUKStochastic is now instantiated with the initial state.