Debugged the initial run on MNIST.