Neural network to classify images of digits