remove prints & fix retrieving on next timestep data