Fixed minibatch FW and added some plots

The plots essentially tell us what minibatch size tau to pick