Add mnist classification example

I noticed it was kind of a struggle to make predictions with the MNIST example. I think I found a pretty easy way to do it and I'm happy to contribute it if you find it useful.