minor update to mnist example file