Update sample.py