Major commit

Version with invariance, MNIST