using MNIST package instead of loading file