Fix batch_reduce so it produces a trajectory return if its a batch of trajs....
Fix batch_reduce so it produces a trajectory return if its a batch of trajs. Also (somewhat) hacky fix for generating a batch when we need two arguments on our function
Closes #15 (closed)