Utilizing the Adam optimizer in our MNIST example