Fixed bug in data generation

numEpochs would calculate to 1 where it should be larger. It now does
what intended, namely calculates numEpochs so that total number of
iterations is at least as many as for when batchSize = 2.

Also tweaked the plots a bit.