Imported from https://github.com/sidSingla/MNIST_Classification