Add exception handling for missing MNIST dataset