update mnist example to proper number of labels