indexes in sample_traj_and_print