Handwritten digit recognition in Keras