Update sample_code.py