Trying to get better reconstruction, not yet there