loop over time steps too