Got the reconstruction error working on MNist. For some reason it does not work on the squares. Also added a .gitignore