Final fixes for MNIST conv nueral net classifier