seperated out mnist and mnist_rotated, added __init__ to make test_mnist importable