MNIST not working well yet...