Reduce Loops over all agents in Tree Obs
Currently in the get() function in TreeObservation we create the lists of all agents position and so on. This should not be computed in the get() but rather the get_many() to avoid computing the same lists multiple times