Added MNIST dataset although results aren't great