Updated mnist example to plot images