edit sample.py