Fix initial stokes solve after loading checkpoint data

If temperature field checkpoint data is loaded a subsequent stokes solving step will be performed for timestep 0. If not, the initial temeprature field will be written out and subseqently a stokes solving step will be performed. This logic is only applied to the timestep 0.