Update MNISTdigits.py

Just a suggestion