Update MNISTdigits.py

Just a suggestion
Merge pull request #5 from Warvito/patch-2

Update MNISTdigits.py