Setting up network parameters in our MNIST example