Use dummy data instead of MNIST