Update download_mnist.py

PEP8 fixes