Basic MNIST image classifier