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)