Update output.py