Rework trajectory to use arbitrary observables by utilizing `observe_state`

[Datseris]

With the new introduction of the observe_state function, we can easily allow any index/observable/symbolic in the save_idxs argument of the trajectory function, allthewhile simplifying its source code as well!