included the MNIST dataset for softmax regression