Merge pull request #3 from jswiergo/mnist-fix

Fixed MNIST stuff