Generalize Data Sets for Training

train_model works on any data set instead of just MNIST digits.

Add a separate script to download the MNIST digits.