Updated mnist example