adding module for MNIST data set