Fixed issue with minibatch size display
Merged code for display issues while printing minibatch size