Update sample_output.py