move the main from convnet to MNIST digits where it belongs