Update file output.py