RNN in Keras for MNIST data