Update reconstruction.py