plotting for mnist