Add a plot of the first MNIST digit