Store subset of variables for trajectories

[patrickfletcher]

Currently stores all variables, which can use a lot of memory. Some use cases only require trajectory of a subset of variables