Add wrapper to help with scanning many timesteps at once