updates to MNIST calassifier