added import for aternate MNIST dataset