Update postprocess.py