Added prediction of MNIST dataset