Implementing a CNN on MNIST digit classification **